Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Ddynamic_dimension_inference_test.cc116 TF_CHECK_OK(module_->dynamic_parameter_binding().Bind( in TEST_F()
135 TF_CHECK_OK(module_->dynamic_parameter_binding().Bind( in TEST_F()
161 TF_CHECK_OK(module_->dynamic_parameter_binding().Bind( in TEST_F()
192 TF_CHECK_OK(module_->dynamic_parameter_binding().Bind( in TEST_F()
223 TF_CHECK_OK(module_->dynamic_parameter_binding().Bind( in TEST_F()
255 TF_CHECK_OK(module_->dynamic_parameter_binding().Bind( in TEST_F()
279 TF_CHECK_OK(module_->dynamic_parameter_binding().Bind( in TEST_F()
332 TF_CHECK_OK(module_->dynamic_parameter_binding().Bind( in TEST_F()
337 TF_CHECK_OK(module_->dynamic_parameter_binding().Bind( in TEST_F()
340 TF_CHECK_OK(module_->dynamic_parameter_binding().Bind( in TEST_F()
[all …]
Ddynamic_padder_test.cc135 TF_CHECK_OK(module_->dynamic_parameter_binding().Bind( in TEST_F()
292 TF_CHECK_OK(module_->dynamic_parameter_binding().Bind( in TEST_F()
335 TF_CHECK_OK(module_->dynamic_parameter_binding().Bind( in TEST_F()
362 TF_CHECK_OK(module_->dynamic_parameter_binding().Bind( in TEST_F()
441 TF_CHECK_OK(module_padded->dynamic_parameter_binding().Bind( in XLA_TEST_F()
444 TF_CHECK_OK(module_padded->dynamic_parameter_binding().Bind( in XLA_TEST_F()
498 TF_CHECK_OK(module_padded->dynamic_parameter_binding().Bind( in XLA_TEST_F()
501 TF_CHECK_OK(module_padded->dynamic_parameter_binding().Bind( in XLA_TEST_F()
605 TF_CHECK_OK(module_padded->dynamic_parameter_binding().Bind( in XLA_TEST_F()
608 TF_CHECK_OK(module_padded->dynamic_parameter_binding().Bind( in XLA_TEST_F()
Dhlo_module.h292 DynamicParameterBinding& dynamic_parameter_binding() { in dynamic_parameter_binding() function
295 const DynamicParameterBinding& dynamic_parameter_binding() const { in dynamic_parameter_binding() function
Ddynamic_dimension_inference.cc1321 DynamicParameterBinding dynamic_parameter_binding; in HandleConditional() local
1330 TF_RETURN_IF_ERROR(dynamic_parameter_binding.Bind(dynamic_parameter, in HandleConditional()
1336 << dynamic_parameter_binding; in HandleConditional()
1338 new_computation, dynamic_parameter_binding, parent_)); in HandleConditional()
1642 VLOG(2) << "Param Config " << module->dynamic_parameter_binding().ToString(); in Run()
1668 module_->entry_computation(), module_->dynamic_parameter_binding(), this, in AnalyzeDynamicDimensions()
Dhlo_module.cc288 dynamic_parameter_binding().ToProto(); in ToProto()
413 proto.dynamic_parameter_binding())); in CreateFromProto()
Dhlo.proto400 DynamicParameterBindingProto dynamic_parameter_binding = 9; field
Ddynamic_padder.cc1794 TF_RETURN_IF_ERROR(module->dynamic_parameter_binding().ForEachBinding( in Run()
Dhlo_verifier.cc1967 TF_RETURN_IF_ERROR(module->dynamic_parameter_binding().Verify(*module)); in Run()
DBUILD411 "dynamic_parameter_binding.cc",
427 "dynamic_parameter_binding.h",
Dhlo_evaluator_test.cc4273 TF_CHECK_OK(m_->dynamic_parameter_binding().Bind( in TEST_F()
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/translate/
Ddynamic_parameter_binding.mlir11 // CHECK: dynamic_parameter_binding
15 // TUPLE: dynamic_parameter_binding
27 // CHECK: dynamic_parameter_binding
34 // TUPLE: dynamic_parameter_binding
50 // CHECK: dynamic_parameter_binding
61 // TUPLE: dynamic_parameter_binding
83 // CHECK: dynamic_parameter_binding
99 // TUPLE: dynamic_parameter_binding
Dsimple.hlo138 dynamic_parameter_binding {
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dmlir_hlo_to_hlo.cc1669 auto* dynamic_parameter_binding = in AddDynamicParameterBindings() local
1744 AddDynamicParameterBindingEntry(dynamic_parameter_binding, i, shape_index, in AddDynamicParameterBindings()