| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| conv_canonicalization.cc | 132 HloInstruction* new_conv = module->entry_computation()->AddInstruction( local
141 hlo->shape(), new_conv,
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| transpose_folding.cc | 177 auto new_conv = HloInstruction::CreateConvolve( local
182 &convolution, std::move(new_conv)));
|
| transpose_folding_test.cc | 264 HloInstruction* new_conv = *instruction_set.begin(); local
265 EXPECT_EQ(HloOpcode::kConvolution, new_conv->opcode());
267 new_conv->convolution_dimension_numbers()
270 new_conv->convolution_dimension_numbers()
320 HloInstruction* new_conv = *instruction_set.begin(); local
321 EXPECT_EQ(HloOpcode::kConvolution, new_conv->opcode());
323 new_conv->convolution_dimension_numbers()
326 new_conv->convolution_dimension_numbers()
330 new_conv->convolution_dimension_numbers().kernel_spatial_dimensions(0));
333 new_conv->convolution_dimension_numbers().kernel_spatial_dimensions(1))
381 HloInstruction* new_conv = *instruction_set.begin(); local
448 HloInstruction* new_conv = *instruction_set.begin(); local
[all...] |
| convolution_group_converter.cc | 529 auto new_conv = HloInstruction::CreateConcatenate( local
532 convolution, std::move(new_conv)));
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| cudnn_conv_pad_for_tensor_cores.cc | 95 auto* new_conv = local
98 // Slice the new conv result if necessary, keeping in mind that new_conv has
107 HloInstruction::CreateGetTupleElement(new_result_shape, new_conv, 0));
112 new_conv =
117 << new_conv->ToString();
118 return conv->parent()->ReplaceInstruction(conv, new_conv);
|
| cudnn_fused_conv_rewriter.cc | 224 auto new_conv = computation->AddInstruction(HloInstruction::CreateCustomCall( local
226 new_conv->set_window(conv->window());
227 new_conv->set_convolution_dimension_numbers(
229 new_conv->set_metadata(conv->metadata());
236 TF_RETURN_IF_ERROR(new_conv->set_backend_config(config));
239 << new_conv->ToString();
241 new_conv, 0);
|
| cudnn_conv_padding_legalization.cc | 183 auto new_conv = conv->parent()->AddInstruction( local
185 new_conv->set_window(new_conv_window);
187 << new_conv->ToString();
188 TF_CHECK_OK(conv->parent()->ReplaceInstruction(conv, new_conv));
|
