HomeSort by relevance Sort by last modified time
    Searched refs:reverse_filter (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
cudnn_convolution_rewriter.cc 223 HloInstruction* reverse_filter = conv->mutable_operand(1); local
228 if (reverse_filter->opcode() == HloOpcode::kReverse) {
229 if (kernel_spatial_dims.size() != reverse_filter->dimensions().size() ||
232 reverse_filter->dimensions().begin())) {
242 << reverse_filter->ToString();
381 if (reverse_filter->opcode() != HloOpcode::kReverse) {
382 reverse_filter = reverse_filter->parent()->AddInstruction(
383 HloInstruction::CreateReverse(reverse_filter->shape(), reverse_filter,
    [all...]
ir_emission_utils.h 110 const Shape& shape, HloInstruction* output, HloInstruction* reverse_filter,
ir_emission_utils.cc 169 const Shape& shape, HloInstruction* output, HloInstruction* reverse_filter,
172 reverse_filter, window, dnums);

Completed in 72 milliseconds