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()
334 std::vector<bool> input_is_dynamic = GetDynamicInputInfo(executable); in DoWork() local
352 input_is_dynamic, release_inputs); in DoWork()
445 std::vector<bool> input_is_dynamic = GetDynamicInputInfo(executable); in DoWork() local
454 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()
305 std::vector<bool> input_is_dynamic = GetDynamicInputInfo(computation_layout); in RunExecutable() local
310 input_tuples, input_is_dynamic, release_inputs)); in RunExecutable()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dreshape_op.cc159 bool input_is_dynamic = false; in Compile() local
166 input_is_dynamic = true; in Compile()
186 dims_are_dynamic[unknown_index] = input_is_dynamic; in Compile()
Dstrided_slice_op.cc333 bool input_is_dynamic = xla_shape.is_dynamic_dimension(input_index); in Compile() local
341 if (input_is_dynamic || end_is_dynamic) { in Compile()
/external/tensorflow/tensorflow/compiler/xrt/
Dxrt_util.cc318 const std::vector<bool>& input_is_dynamic, bool release_inputs) { in GetArgumentsBuffers() argument
320 return arg < input_is_dynamic.size() && input_is_dynamic[arg]; in GetArgumentsBuffers()
Dxrt_util.h92 const std::vector<bool>& input_is_dynamic, bool release_inputs);
/external/tensorflow/tensorflow/compiler/xla/service/
Ddynamic_padder.cc2032 const bool input_is_dynamic = absl::c_any_of( in DefaultAction() local
2051 if (!input_is_dynamic && op_support == OpDynamismSupport::kNoSupport) { in DefaultAction()
2058 if (input_is_dynamic && op_support == OpDynamismSupport::kNoSupport) { in DefaultAction()
2074 if (!input_is_dynamic && op_support == OpDynamismSupport::kRequired) { in DefaultAction()