Searched refs:kConvolution (Results 1 - 15 of 15) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | thunk.h | 45 kConvolution, member in class:xla::gpu::Thunk::Kind
|
| H A D | cudnn_convolution_rewriter.cc | 57 // Precondition: "conv" is a kConvolution.
76 CHECK_EQ(HloOpcode::kConvolution, conv->opcode());
202 // semantics of kConvolution). The batch dimension of the activations should
215 // Precondition: "conv" is a kConvolution.
222 CHECK_EQ(HloOpcode::kConvolution, conv->opcode());
397 CHECK_EQ(conv->opcode(), HloOpcode::kConvolution);
449 if (hlo->opcode() == HloOpcode::kConvolution) {
|
| H A D | convolution_thunk.cc | 49 : Thunk(Kind::kConvolution, hlo),
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| H A D | conv_canonicalization.cc | 35 if (hlo->opcode() == HloOpcode::kConvolution &&
|
| H A D | conv_canonicalization_test.cc | 98 EXPECT_EQ(HloOpcode::kConvolution, canonical_conv->opcode());
|
| H A D | cpu_layout_assignment.cc | 102 if (instruction->opcode() == HloOpcode::kConvolution &&
|
| H A D | parallel_task_assignment.cc | 141 (instruction->opcode() == HloOpcode::kConvolution &&
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | hlo_opcode.h | 65 V(kConvolution, "convolution") \
|
| H A D | transpose_folding.cc | 57 if (HloOpcode::kConvolution != convolution.opcode()) {
|
| H A D | transpose_folding_test.cc | 257 EXPECT_EQ(HloOpcode::kConvolution, new_conv->opcode());
310 EXPECT_EQ(HloOpcode::kConvolution, new_conv->opcode());
368 EXPECT_EQ(HloOpcode::kConvolution, new_conv->opcode());
432 EXPECT_EQ(HloOpcode::kConvolution, new_conv->opcode());
|
| H A D | liveness_util.cc | 218 return operand->opcode() == HloOpcode::kConvolution ||
323 return operand->opcode() == HloOpcode::kConvolution ||
|
| H A D | instruction_fusion.cc | 97 case HloOpcode::kConvolution:
|
| H A D | hlo_instruction.cc | 326 WrapUnique(new HloInstruction(HloOpcode::kConvolution, shape));
1291 case HloOpcode::kConvolution:
1770 case HloOpcode::kConvolution:
2398 if (opcode() == HloOpcode::kConvolution) {
2410 // "convolution" so they match the categories of proper kDot and kConvolution
2586 case HloOpcode::kConvolution:
3329 // A convolution can be represented as a kConvolution HLO or as a CustomCall
|
| H A D | hlo_graph_dumper.cc | 963 case HloOpcode::kConvolution:
|
| /external/tensorflow/tensorflow/compiler/xla/tools/parser/ |
| H A D | hlo_parser.cc | 684 case HloOpcode::kConvolution: {
|
Completed in 366 milliseconds