Searched refs:dynamic_dimension_inference (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_get_dimension_size_rewriter.cc | 31 const DynamicDimensionInference* dynamic_dimension_inference) { in ReplaceGetSize() argument
45 dynamic_dimension_inference->GetDynamicSize(operand, {}, dim); in ReplaceGetSize()
|
| D | dynamic_padder.cc | 111 TF_ASSIGN_OR_RETURN(DynamicDimensionInference dynamic_dimension_inference, in Run()
124 dynamic_dimension_inference.GetDynamicSize(operand, {}, dim); in Run()
|
| D | hlo_evaluator.h | 130 DynamicDimensionInference* dynamic_dimension_inference) { in set_dynamic_dimension_inference() argument
131 dynamic_dimension_inference_ = dynamic_dimension_inference; in set_dynamic_dimension_inference()
|
| D | BUILD | 260 ":dynamic_dimension_inference",
725 ":dynamic_dimension_inference",
2019 name = "dynamic_dimension_inference",
2020 srcs = ["dynamic_dimension_inference.cc"],
2021 hdrs = ["dynamic_dimension_inference.h"],
2040 ":dynamic_dimension_inference",
2081 ":dynamic_dimension_inference",
3110 ":dynamic_dimension_inference",
|
| D | service.cc | 1100 TF_ASSIGN_OR_RETURN(DynamicDimensionInference dynamic_dimension_inference, in ComputeConstantGraph()
1104 evaluator.set_dynamic_dimension_inference(&dynamic_dimension_inference); in ComputeConstantGraph()
|
| D | hlo_evaluator_test.cc | 3050 TF_ASSERT_OK_AND_ASSIGN(DynamicDimensionInference dynamic_dimension_inference, in TEST_F()
3053 evaluator_.set_dynamic_dimension_inference(&dynamic_dimension_inference); in TEST_F()
|