Home
last modified time | relevance | path

Searched refs:dynamic_parameter_binding (Results 1 – 11 of 11) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Ddynamic_dimension_inference_test.cc123 TF_CHECK_OK(module_->dynamic_parameter_binding().Bind( in TEST_F()
142 TF_CHECK_OK(module_->dynamic_parameter_binding().Bind( in TEST_F()
168 TF_CHECK_OK(module_->dynamic_parameter_binding().Bind( in TEST_F()
199 TF_CHECK_OK(module_->dynamic_parameter_binding().Bind( in TEST_F()
230 TF_CHECK_OK(module_->dynamic_parameter_binding().Bind( in TEST_F()
262 TF_CHECK_OK(module_->dynamic_parameter_binding().Bind( in TEST_F()
286 TF_CHECK_OK(module_->dynamic_parameter_binding().Bind( in TEST_F()
339 TF_CHECK_OK(module_->dynamic_parameter_binding().Bind( in TEST_F()
344 TF_CHECK_OK(module_->dynamic_parameter_binding().Bind( in TEST_F()
347 TF_CHECK_OK(module_->dynamic_parameter_binding().Bind( in TEST_F()
[all …]
H A Ddynamic_padder_test.cc177 TF_CHECK_OK(module_->dynamic_parameter_binding().Bind( in TEST_F()
364 TF_CHECK_OK(module_->dynamic_parameter_binding().Bind( in TEST_F()
407 TF_CHECK_OK(module_->dynamic_parameter_binding().Bind( in TEST_F()
434 TF_CHECK_OK(module_->dynamic_parameter_binding().Bind( in TEST_F()
472 TF_CHECK_OK(module_->dynamic_parameter_binding().Bind( in TEST_F()
599 TF_CHECK_OK(module_padded->dynamic_parameter_binding().Bind( in XLA_TEST_F()
602 TF_CHECK_OK(module_padded->dynamic_parameter_binding().Bind( in XLA_TEST_F()
697 TF_CHECK_OK(module_padded->dynamic_parameter_binding().Bind( in XLA_TEST_F()
700 TF_CHECK_OK(module_padded->dynamic_parameter_binding().Bind( in XLA_TEST_F()
804 TF_CHECK_OK(module_padded->dynamic_parameter_binding().Bind( in XLA_TEST_F()
[all …]
H A Dhlo_module.h390 DynamicParameterBinding& dynamic_parameter_binding() { in dynamic_parameter_binding() function
393 const DynamicParameterBinding& dynamic_parameter_binding() const { in dynamic_parameter_binding() function
H A Ddynamic_dimension_inference.cc1496 DynamicParameterBinding dynamic_parameter_binding; in HandleConditional() local
1505 TF_RETURN_IF_ERROR(dynamic_parameter_binding.Bind(dynamic_parameter, in HandleConditional()
1511 << dynamic_parameter_binding; in HandleConditional()
1513 new_computation, dynamic_parameter_binding, parent_)); in HandleConditional()
1949 VLOG(2) << "Param Config " << module->dynamic_parameter_binding().ToString(); in Run()
1980 module_->entry_computation(), module_->dynamic_parameter_binding(), this, in AnalyzeDynamicDimensions()
H A Dhlo_module.cc320 dynamic_parameter_binding().ToProto(); in ToProto()
471 proto.dynamic_parameter_binding())); in CreateFromProto()
H A Dhlo.proto483 DynamicParameterBindingProto dynamic_parameter_binding = 9; field
H A Ddynamic_padder.cc2144 TF_RETURN_IF_ERROR(module->dynamic_parameter_binding().ForEachBinding( in Run()
H A Dhlo_verifier.cc2585 TF_RETURN_IF_ERROR(module->dynamic_parameter_binding().Verify(*module)); in Run()
H A DBUILD573 "dynamic_parameter_binding.cc",
589 "dynamic_parameter_binding.h",
H A Dhlo_evaluator_test.cc4371 TF_CHECK_OK(m_->dynamic_parameter_binding().Bind( in TEST_F()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/tests/translate/
H A Dsimple.hlo138 dynamic_parameter_binding {