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) { in GetInputTuples() 300 bool release_inputs, se::Stream* stream, int rng_seed, in RunExecutable() 379 bool release_inputs, se::Stream* stream, int rng_seed, in ExecuteComputation() 398 const std::vector<InputCoords>& input_coords, bool release_inputs, in ExecuteComputation() 454 bool release_inputs = config_proto.release_input_handles(); in DoWork() local
|
D | tpu_execute_op.cc | 304 bool release_inputs = config_proto.release_input_handles(); in DoWork() local
|
/external/tensorflow/tensorflow/compiler/xrt/ |
D | xrt_util.cc | 259 const std::function<xla::Shape(int64)>& shape_getter, bool release_inputs) { in GetInputTupleAllocations() 315 const std::vector<bool>& input_is_dynamic, bool release_inputs) { in GetArgumentsBuffers()
|