Searched defs:captured_inputs (Results 1 - 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/data/
H A Dcaptured_function.h43 // 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 Dcaptured_function.cc29 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 Dfunction.py313 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 Dfunction.py473 def captured_inputs(self): member in class:GraphModeFunction

Completed in 867 milliseconds