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

  /external/tensorflow/tensorflow/compiler/xla/service/cpu/
conv_canonicalization_test.cc 41 dim->set_stride(1);
  /external/tensorflow/tensorflow/compiler/xla/service/
transpose_folding_test.cc 238 dim->set_stride(1);
294 dim->set_stride(1);
356 dim->set_stride(1);
423 dim->set_stride(1);
convolution_group_converter.cc 293 dim->set_stride(batch_group_count);
296 dim->set_stride(1);
439 dim->set_stride(1);
dynamic_dimension_inference_test.cc 560 batch_dim->set_stride(1);
570 dim->set_stride(2);
610 batch_dim->set_stride(1);
620 dim->set_stride(2);
hlo_evaluator_test.cc     [all...]
shape_inference_test.cc 81 dim.set_stride(2);
307 dim.set_stride(2);
415 dim0->set_stride(2);
421 dim1->set_stride(1);
459 dim0->set_stride(3);
467 dim1->set_stride(1);
505 dim0->set_stride(3);
513 dim1->set_stride(1);
550 dim0->set_stride(1);
554 dim1->set_stride(2)
    [all...]
algebraic_simplifier_test.cc     [all...]
bfloat16_propagation_test.cc 165 dim.set_stride(1);
    [all...]
hlo_instructions.cc     [all...]
hlo_parser.cc     [all...]
  /external/tensorflow/tensorflow/core/kernels/
cudnn_pooling_gpu.cc 84 pooling_desc.set_stride(dim_i, stride[i]);
205 pooling_desc.set_stride(dim_i, stride[i]);
  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
cudnn_conv_rewriter.cc 170 dim->set_stride(conv->window().dimensions(i).window_dilation());
343 dim->set_stride(old_window.dimensions(i).base_dilation());
cudnn_conv_rewriter_test.cc 45 window_dim->set_stride(1);
  /external/tensorflow/tensorflow/compiler/xla/
window_util.cc 34 dimension->set_stride(1);
reference_util.cc 514 dim.set_stride(kernel_stride.first);
524 dim2.set_stride(kernel_stride.second);
  /external/tensorflow/tensorflow/compiler/xla/client/
xla_builder.cc 736 slice_config->set_stride(strides[i]);
    [all...]
  /external/tensorflow/tensorflow/stream_executor/
dnn.h 696 PoolingDescriptor& set_stride(DimIndex dim, int64 value) { function in class:stream_executor::dnn::PoolingDescriptor
    [all...]

Completed in 1026 milliseconds