Home
last modified time | relevance | path

Searched refs:dynamic_dimension_inference (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Ddynamic_padder.cc155 DynamicDimensionInference* dynamic_dimension_inference) { in ReplaceGetSize() argument
171 dynamic_dimension_inference->GetDynamicSize(operand, {}, dim); in ReplaceGetSize()
177 dynamic_dimension_inference->ReplaceAllDynamicDimensionUsesWith( in ReplaceGetSize()
184 dynamic_dimension_inference->ReplaceAllDynamicDimensionUsesWith(instr, in ReplaceGetSize()
344 DynamicDimensionInference* dynamic_dimension_inference) { in RewriteDynamicReshapeSplitInput() argument
474 dynamic_dimension_inference->GetDynamicSize(reshape, {}, output_dim); in RewriteDynamicReshapeSplitInput()
486 TF_RETURN_IF_ERROR(dynamic_dimension_inference->ForwardDynamicSize( in RewriteDynamicReshapeSplitInput()
558 DynamicDimensionInference* dynamic_dimension_inference) { in RewriteDynamicReshapeCombineInput() argument
682 dynamic_dimension_inference->GetDynamicSize(reshape, {}, output_dim); in RewriteDynamicReshapeCombineInput()
699 dynamic_dimension_inference->ForwardDynamicSize(reshape, gather, {})); in RewriteDynamicReshapeCombineInput()
[all …]
Dhlo_evaluator.h132 DynamicDimensionInference* dynamic_dimension_inference) { in set_dynamic_dimension_inference() argument
133 dynamic_dimension_inference_ = dynamic_dimension_inference; in set_dynamic_dimension_inference()
136 DynamicDimensionInference* dynamic_dimension_inference() { in dynamic_dimension_inference() function
Dservice.cc1077 TF_ASSIGN_OR_RETURN(DynamicDimensionInference dynamic_dimension_inference, in ComputeConstantGraph()
1081 evaluator.set_dynamic_dimension_inference(&dynamic_dimension_inference); in ComputeConstantGraph()
DBUILD290 ":dynamic_dimension_inference",
944 ":dynamic_dimension_inference",
2777 name = "dynamic_dimension_inference",
2778 srcs = ["dynamic_dimension_inference.cc"],
2779 hdrs = ["dynamic_dimension_inference.h"],
2807 ":dynamic_dimension_inference",
2866 ":dynamic_dimension_inference",
Dhlo_evaluator_test.cc4277 TF_ASSERT_OK_AND_ASSIGN(DynamicDimensionInference dynamic_dimension_inference, in TEST_F()
4280 evaluator_.set_dynamic_dimension_inference(&dynamic_dimension_inference); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/interpreter/
Dcompiler.cc111 TF_ASSIGN_OR_RETURN(DynamicDimensionInference dynamic_dimension_inference, in RunBackend()
123 std::move(dynamic_dimension_inference)); in RunBackend()
DBUILD87 "//tensorflow/compiler/xla/service:dynamic_dimension_inference",