| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | convolution_4d_expander_test.cc | 46 EXPECT_EQ(root->opcode(), HloOpcode::kConvolution); in TEST_F()
54 EXPECT_EQ(new_convolution->opcode(), HloOpcode::kConvolution); in TEST_F()
71 EXPECT_EQ(root->opcode(), HloOpcode::kConvolution); in TEST_F()
81 EXPECT_EQ(new_convolution->opcode(), HloOpcode::kConvolution); in TEST_F()
98 EXPECT_EQ(root->opcode(), HloOpcode::kConvolution); in TEST_F()
106 EXPECT_EQ(new_convolution->opcode(), HloOpcode::kConvolution); in TEST_F()
123 EXPECT_EQ(root->opcode(), HloOpcode::kConvolution); in TEST_F()
142 EXPECT_EQ(root->opcode(), HloOpcode::kConvolution); in TEST_F()
161 EXPECT_EQ(root->opcode(), HloOpcode::kConvolution); in TEST_F()
|
| D | convolution_group_converter_test.cc | 50 EXPECT_EQ(root->opcode(), HloOpcode::kConvolution); in TEST_F()
57 EXPECT_EQ(root->opcode(), HloOpcode::kConvolution); in TEST_F()
82 EXPECT_EQ(root->opcode(), HloOpcode::kConvolution); in TEST_F()
90 EXPECT_EQ(root->operand(0)->opcode(), HloOpcode::kConvolution); in TEST_F()
109 EXPECT_EQ(root->opcode(), HloOpcode::kConvolution); in TEST_F()
136 EXPECT_EQ(root->opcode(), HloOpcode::kConvolution); in TEST_F()
|
| D | transpose_folding_test.cc | 267 EXPECT_EQ(HloOpcode::kConvolution, new_conv->opcode()); in TEST_F()
324 EXPECT_EQ(HloOpcode::kConvolution, new_conv->opcode()); in TEST_F()
386 EXPECT_EQ(HloOpcode::kConvolution, new_conv->opcode()); in TEST_F()
454 EXPECT_EQ(HloOpcode::kConvolution, new_conv->opcode()); in TEST_F()
|
| D | convert_operand_folding.cc | 37 instruction->opcode() != HloOpcode::kConvolution) { in InstructionMatchesPattern()
|
| D | operand_upcaster.cc | 31 case HloOpcode::kConvolution: in MaybeInferShape()
|
| D | dot_as_convolution_util.cc | 54 CHECK_EQ(conv->opcode(), HloOpcode::kConvolution); in ParseConvolutionDimsInfo()
112 CHECK_EQ(conv.opcode(), HloOpcode::kConvolution); in CreateShardedConvForDotGeneralConvolution()
|
| D | hlo_opcode.h | 75 V(kConvolution, "convolution", 2) \
|
| D | convolution_4d_expander.cc | 33 if (instruction->opcode() != HloOpcode::kConvolution) { in InstructionMatchesPattern()
|
| D | hlo_cse.cc | 132 case HloOpcode::kConvolution: in CseHash()
|
| D | transpose_folding.cc | 89 if (HloOpcode::kConvolution != convolution.opcode()) { in CanFoldOperandsIntoConvolution()
|
| D | sharding_propagation.cc | 130 CHECK_EQ(instruction->opcode(), HloOpcode::kConvolution); in IsConvolutionKernelSmall()
243 case HloOpcode::kConvolution: in PickRepresentativeOperand()
303 case HloOpcode::kConvolution: in SupportSpatialPartitioning()
776 case HloOpcode::kConvolution: in InferShardingFromOperands()
1202 case HloOpcode::kConvolution: { in GetShardingFromUser()
|
| D | space_to_batch_converter.cc | 258 if (inst->opcode() != HloOpcode::kConvolution) { in ConvolutionVisitor()
632 if (instr->opcode() == HloOpcode::kConvolution) { in Run()
636 if (instr->opcode() == HloOpcode::kConvolution && in Run()
756 if (consumer->opcode() == HloOpcode::kConvolution) { in CanPropagate()
1125 if (consumer->opcode() == HloOpcode::kConvolution) { in SupportedOpForPropagation()
1351 if (consumer->opcode() == HloOpcode::kConvolution) { in Propagate()
2577 if (user->opcode() == HloOpcode::kConvolution || in DoesConvolutionFeedReduceWindowOrSelectAndScatter()
|
| D | hlo_replication_analysis.cc | 141 hlo->opcode() == HloOpcode::kConvolution || // in DetermineHloInstructionIsReplicated()
|
| D | instruction_fusion.cc | 149 case HloOpcode::kConvolution: in IsExpensive()
|
| D | dynamic_padder.cc | 105 case HloOpcode::kConvolution: in ChooseIdentityValue()
229 if (operand_num == 0 && inst->opcode() == HloOpcode::kConvolution && in ShouldSkipPadOnOperand()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | thunk.cc | 41 case Thunk::kConvolution: in ThunkKindToString()
|
| D | convolution_thunk.cc | 37 : Thunk(Kind::kConvolution, thunk_info), in ConvolutionThunk()
|
| D | gpu_conv_rewriter.cc | 176 CHECK_EQ(HloOpcode::kConvolution, conv->opcode()); in MatchBackwardFilter()
342 CHECK_EQ(HloOpcode::kConvolution, conv->opcode()); in MatchBackwardInput()
693 CHECK_EQ(conv->opcode(), HloOpcode::kConvolution); in RunOnInstruction()
721 if (hlo->opcode() == HloOpcode::kConvolution) { in RunOnComputation()
|
| D | thunk.h | 50 kConvolution, enumerator
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | convolution_handler.cc | 49 TF_RET_CHECK(original_hlo->opcode() == HloOpcode::kConvolution); in PartitionConvolutionWithBatchGroupCount()
141 TF_RET_CHECK(original_hlo->opcode() == HloOpcode::kConvolution); in PartitionConvolutionWithFeatureGroupCount()
235 TF_RET_CHECK(original_hlo->opcode() == HloOpcode::kConvolution); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()
522 TF_RET_CHECK(original_hlo->opcode() == HloOpcode::kConvolution); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS()
747 TF_RET_CHECK(original_hlo->opcode() == HloOpcode::kConvolution); in PartitionConvolutionTiledOutput()
825 TF_RET_CHECK(original_hlo->opcode() == HloOpcode::kConvolution); in PartitionConvolutionBaseCase()
898 CHECK_EQ(conv.opcode(), HloOpcode::kConvolution); in CreateShardedConvConvolution()
973 TF_RET_CHECK(original_hlo->opcode() == HloOpcode::kConvolution); in PartitionConvolution()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_layout_assignment.cc | 99 if (instr.opcode() == HloOpcode::kConvolution) { in OperandsAndResultMustHaveRowMajorLayout()
|
| D | conv_canonicalization_test.cc | 104 EXPECT_EQ(HloOpcode::kConvolution, canonical_conv->opcode()); in TEST_F()
|
| D | conv_canonicalization.cc | 36 if (hlo->opcode() == HloOpcode::kConvolution && in Run()
|
| D | parallel_task_assignment.cc | 161 (opcode == HloOpcode::kConvolution && in GetTargetParallelTaskCount()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | hlo_utils.cc | 360 return xla::HloOpcode::kConvolution; in MhloToHloOpcode()
|