Searched refs:GetDynamicSize (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | dynamic_dimension_inference_test.cc | 98 EXPECT_EQ(inference_->GetDynamicSize(param, {}, 1), param2); in TEST_F()
99 EXPECT_EQ(inference_->GetDynamicSize(param, {}, 0), nullptr); in TEST_F()
100 EXPECT_EQ(inference_->GetDynamicSize(param2, {}, 0), nullptr); in TEST_F()
118 EXPECT_THAT(inference_->GetDynamicSize(param, {0}, 1), in TEST_F()
121 EXPECT_EQ(inference_->GetDynamicSize(param, {0}, 0), nullptr); in TEST_F()
144 EXPECT_THAT(inference_->GetDynamicSize(param, {0}, 1), in TEST_F()
147 EXPECT_THAT(inference_->GetDynamicSize(gte, {}, 1), in TEST_F()
150 EXPECT_EQ(inference_->GetDynamicSize(param, {0}, 0), nullptr); in TEST_F()
175 EXPECT_EQ(inference_->GetDynamicSize(negate, {}, 1), size_param); in TEST_F()
206 EXPECT_EQ(inference_->GetDynamicSize(reduce, {}, 0), size_param); in TEST_F()
[all …]
|
| D | hlo_get_dimension_size_rewriter.cc | 45 dynamic_dimension_inference->GetDynamicSize(operand, {}, dim); in ReplaceGetSize()
|
| D | dynamic_dimension_inference.h | 46 HloInstruction* GetDynamicSize(HloInstruction* inst, const ShapeIndex& index,
|
| D | dynamic_dimension_inference.cc | 563 HloInstruction* dynamic_size = parent_->GetDynamicSize( in ForEachOperandDynamicDimension()
581 GetDynamicSize(dynamic_dimension.inst, dynamic_dimension.index, in CopyMapping()
620 HloInstruction* DynamicDimensionInference::GetDynamicSize( in GetDynamicSize() function in xla::DynamicDimensionInference
|
| D | dynamic_padder.cc | 124 dynamic_dimension_inference.GetDynamicSize(operand, {}, dim); in Run()
|
| D | hlo_evaluator.cc | 402 dynamic_dimension_inference_->GetDynamicSize(operand, {}, dim); in HandleGetDimensionSize()
|