Searched refs:dynamic_dimension_inference (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | dynamic_padder.cc | 167 DynamicDimensionInference* dynamic_dimension_inference) { in ReplaceGetSize() argument
183 dynamic_dimension_inference->GetDynamicSize(operand, {}, dim); in ReplaceGetSize()
189 dynamic_dimension_inference->ReplaceAllDynamicDimensionUsesWith( in ReplaceGetSize()
196 dynamic_dimension_inference->ReplaceAllDynamicDimensionUsesWith(instr, in ReplaceGetSize()
472 DynamicDimensionInference* dynamic_dimension_inference) { in RewriteDynamicReshapeSplitInput() argument
575 dynamic_dimension_inference->GetDynamicSize(reshape, {}, output_dim); in RewriteDynamicReshapeSplitInput()
587 TF_RETURN_IF_ERROR(dynamic_dimension_inference->ForwardDynamicSize( in RewriteDynamicReshapeSplitInput()
654 DynamicDimensionInference* dynamic_dimension_inference) { in RewriteDynamicReshapeCombineInput() argument
747 dynamic_dimension_inference->GetDynamicSize(reshape, {}, output_dim); in RewriteDynamicReshapeCombineInput()
764 dynamic_dimension_inference->ForwardDynamicSize(reshape, gather, {})); in RewriteDynamicReshapeCombineInput()
[all …]
|
| D | hlo_evaluator.h | 189 DynamicDimensionInference* dynamic_dimension_inference) { in set_dynamic_dimension_inference() argument
190 dynamic_dimension_inference_ = dynamic_dimension_inference; in set_dynamic_dimension_inference()
193 DynamicDimensionInference* dynamic_dimension_inference() { in dynamic_dimension_inference() function
|
| D | service.cc | 1147 TF_ASSIGN_OR_RETURN(DynamicDimensionInference dynamic_dimension_inference, in ComputeConstantGraph()
1151 evaluator.set_dynamic_dimension_inference(&dynamic_dimension_inference); in ComputeConstantGraph()
|
| D | BUILD | 452 ":dynamic_dimension_inference",
1143 ":dynamic_dimension_inference",
3596 name = "dynamic_dimension_inference",
3597 srcs = ["dynamic_dimension_inference.cc"],
3598 hdrs = ["dynamic_dimension_inference.h"],
3665 ":dynamic_dimension_inference",
3694 ":dynamic_dimension_inference",
3728 ":dynamic_dimension_inference",
|
| D | hlo_evaluator_test.cc | 4375 TF_ASSERT_OK_AND_ASSIGN(DynamicDimensionInference dynamic_dimension_inference, in TEST_F()
4378 evaluator_.set_dynamic_dimension_inference(&dynamic_dimension_inference); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| D | compiler.cc | 116 TF_ASSIGN_OR_RETURN(DynamicDimensionInference dynamic_dimension_inference, in RunBackend()
128 std::move(dynamic_dimension_inference)); in RunBackend()
|
| D | BUILD | 88 "//tensorflow/compiler/xla/service:dynamic_dimension_inference",
|