Searched refs:release_inputs (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | xrt_execute_op.cc | 91 bool release_inputs) { in GetInputTuples() argument
98 release_inputs); in GetInputTuples()
300 bool release_inputs, se::Stream* stream, int rng_seed, in RunExecutable() argument
309 input_tuples, input_is_dynamic, release_inputs)); in RunExecutable()
379 bool release_inputs, se::Stream* stream, int rng_seed, in ExecuteComputation() argument
383 release_inputs, stream, rng_seed, config); in ExecuteComputation()
398 const std::vector<InputCoords>& input_coords, bool release_inputs, in ExecuteComputation() argument
405 input_coords, release_inputs)); in ExecuteComputation()
407 executable, input_tuples, release_inputs, stream, in ExecuteComputation()
454 bool release_inputs = config_proto.release_input_handles(); in DoWork() local
[all …]
|
| D | tpu_execute_op.cc | 304 bool release_inputs = config_proto.release_input_handles(); in DoWork() local
347 release_inputs)); in DoWork()
350 input_is_dynamic, release_inputs); in DoWork()
|
| /external/tensorflow/tensorflow/compiler/xrt/ |
| D | xrt_util.h | 81 const std::function<xla::Shape(int64)>& shape_getter, bool release_inputs);
91 const std::vector<bool>& input_is_dynamic, bool release_inputs);
|
| D | xrt_util.cc | 259 const std::function<xla::Shape(int64)>& shape_getter, bool release_inputs) { in GetInputTupleAllocations() argument
271 if (release_inputs) { in GetInputTupleAllocations()
315 const std::vector<bool>& input_is_dynamic, bool release_inputs) { in GetArgumentsBuffers() argument
326 bool alias_outputs = release_inputs && input_tuples.size() == 1 && in GetArgumentsBuffers()
|