Searched defs:release_inputs (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | xrt_execute_op.cc | 91 bool release_inputs, se::DeviceMemoryAllocator* allocator) { in GetInputTuples()
301 bool release_inputs, se::Stream* stream, int rng_seed, in RunExecutable()
383 bool release_inputs, se::Stream* stream, int rng_seed, in ExecuteComputation()
402 const std::vector<InputCoords>& input_coords, bool release_inputs, in ExecuteComputation()
458 bool release_inputs = config_proto.release_input_handles(); in DoWork() local
|
| D | tpu_execute_op.cc | 306 bool release_inputs = config_proto.release_input_handles(); in DoWork() local
|
| /external/tensorflow/tensorflow/compiler/xrt/ |
| D | xrt_util.cc | 261 const std::function<xla::Shape(int64_t)>& shape_getter, bool release_inputs, in GetInputTupleAllocations()
318 const std::vector<bool>& input_is_dynamic, bool release_inputs) { in GetArgumentsBuffers()
|