Home
last modified time | relevance | path

Searched refs:input_is_dynamic (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/xrt/kernels/
Dtpu_execute_op.cc76 std::vector<bool> input_is_dynamic; in GetDynamicInputInfo() local
77 input_is_dynamic.reserve(executable_proto.input_shapes().size()); in GetDynamicInputInfo()
79 input_is_dynamic.push_back( in GetDynamicInputInfo()
82 return input_is_dynamic; in GetDynamicInputInfo()
332 std::vector<bool> input_is_dynamic = GetDynamicInputInfo(executable); in DoWork() local
350 input_is_dynamic, release_inputs); in DoWork()
443 std::vector<bool> input_is_dynamic = GetDynamicInputInfo(executable); in DoWork() local
452 input_is_dynamic, in DoWork()
Dxrt_execute_op.cc79 std::vector<bool> input_is_dynamic; in GetDynamicInputInfo() local
80 input_is_dynamic.reserve(computation_layout.parameter_count()); in GetDynamicInputInfo()
82 input_is_dynamic.push_back( in GetDynamicInputInfo()
85 return input_is_dynamic; in GetDynamicInputInfo()
304 std::vector<bool> input_is_dynamic = GetDynamicInputInfo(computation_layout); in RunExecutable() local
309 input_tuples, input_is_dynamic, release_inputs)); in RunExecutable()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dreshape_op.cc141 bool input_is_dynamic = false; in Compile() local
148 input_is_dynamic = true; in Compile()
168 dims_are_dynamic[unknown_index] = input_is_dynamic; in Compile()
Dstrided_slice_op.cc296 bool input_is_dynamic = xla_shape.is_dynamic_dimension(input_index); in Compile() local
304 if ((input_is_dynamic && backward_slice) || end_is_dynamic) { in Compile()
/external/tensorflow/tensorflow/compiler/xrt/
Dxrt_util.cc315 const std::vector<bool>& input_is_dynamic, bool release_inputs) { in GetArgumentsBuffers() argument
317 return arg < input_is_dynamic.size() && input_is_dynamic[arg]; in GetArgumentsBuffers()
Dxrt_util.h91 const std::vector<bool>& input_is_dynamic, bool release_inputs);
/external/tensorflow/tensorflow/compiler/xla/service/
Ddynamic_padder.cc1686 const bool input_is_dynamic = absl::c_any_of( in DefaultAction() local
1705 if (!input_is_dynamic && op_support == OpDynamismSupport::kNoSupport) { in DefaultAction()
1712 if (input_is_dynamic && op_support == OpDynamismSupport::kNoSupport) { in DefaultAction()
1727 if (!input_is_dynamic && op_support == OpDynamismSupport::kRequired) { in DefaultAction()