Searched defs:captured_inputs (Results 1 - 4 of 4) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/data/ |
H A D | captured_function.h | 43 // NOTE(mrry): The `captured_inputs` are passed by value. For 46 std::vector<Tensor> captured_inputs, 77 const std::vector<Tensor>& captured_inputs() { return captured_inputs_; } function in class:tensorflow::CapturedFunction 91 std::vector<Tensor> captured_inputs);
|
H A D | captured_function.cc | 29 const NameAttrList& func, std::vector<Tensor> captured_inputs, 31 out_function->reset(new CapturedFunction(func, std::move(captured_inputs))); 92 const std::vector<Tensor>* captured_inputs, 96 captured_inputs_(captured_inputs) {} 128 const std::vector<Tensor>* captured_inputs, 132 captured_inputs_(captured_inputs) {} 312 std::vector<Tensor> captured_inputs) 316 captured_inputs_(std::move(captured_inputs)) {} 28 Create( const NameAttrList& func, std::vector<Tensor> captured_inputs, std::unique_ptr<CapturedFunction>* out_function) argument 91 OwnedArgsCallFrame(std::vector<Tensor>&& args, const std::vector<Tensor>* captured_inputs, DataTypeSlice ret_types) argument 127 BorrowedArgsCallFrame(const std::vector<Tensor>& args, const std::vector<Tensor>* captured_inputs, DataTypeSlice ret_types) argument 311 CapturedFunction(const NameAttrList& func, std::vector<Tensor> captured_inputs) argument
|
/external/tensorflow/tensorflow/python/framework/ |
H A D | function.py | 313 def captured_inputs(self): member in class:_DefinedFunction 723 captured_inputs = [self._add_tensor_and_parents(x) for x in op.inputs] 727 captured_inputs, [o.dtype for o in op.outputs],
|
/external/tensorflow/tensorflow/python/eager/ |
H A D | function.py | 473 def captured_inputs(self): member in class:GraphModeFunction
|
Completed in 867 milliseconds