| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| 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) {
|
| convolution_thunk.cc | 49 : Thunk(Kind::kConvolution, hlo),
|
| ir_emission_utils.h | 77 // CudnnConvolutionRewriter lowers kConvolution HLOs to these custom calls.
96 // kConvolution opcode.
|
| thunk.h | 45 kConvolution,
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| 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());
|
| hlo_opcode.h | 65 V(kConvolution, "convolution") \
|
| transpose_folding.cc | 57 if (HloOpcode::kConvolution != convolution.opcode()) {
|
| liveness_util.cc | 218 return operand->opcode() == HloOpcode::kConvolution ||
323 return operand->opcode() == HloOpcode::kConvolution ||
|
| instruction_fusion.cc | 97 case HloOpcode::kConvolution:
|
| hlo_instruction.cc | 326 WrapUnique(new HloInstruction(HloOpcode::kConvolution, shape));
[all...] |
| hlo_graph_dumper.cc | [all...] |
| hlo_instruction.h | [all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| conv_canonicalization.cc | 35 if (hlo->opcode() == HloOpcode::kConvolution &&
|
| conv_canonicalization_test.cc | 98 EXPECT_EQ(HloOpcode::kConvolution, canonical_conv->opcode());
|
| cpu_layout_assignment.cc | 102 if (instruction->opcode() == HloOpcode::kConvolution &&
|
| parallel_task_assignment.cc | 141 (instruction->opcode() == HloOpcode::kConvolution &&
|
| /external/tensorflow/tensorflow/compiler/xla/tools/parser/ |
| hlo_parser.cc | 684 case HloOpcode::kConvolution: {
[all...] |
