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

  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
cudnn_convolution_runner.cc 145 .set_zero_padding(
169 convolution_descriptor.set_zero_padding(static_cast<DimIndex>(0), 0)
  /external/tensorflow/tensorflow/core/kernels/
conv_ops_3d.cc 336 .set_zero_padding(DimIndex::X, pad_cols / 2)
337 .set_zero_padding(DimIndex::Y, pad_rows / 2)
338 .set_zero_padding(DimIndex::Z, pad_planes / 2);
conv_grad_ops_3d.cc 586 .set_zero_padding(DimIndex::X, padding_cols / 2)
587 .set_zero_padding(DimIndex::Y, padding_rows / 2)
588 .set_zero_padding(DimIndex::Z, padding_planes / 2);
    [all...]
  /external/tensorflow/tensorflow/stream_executor/
dnn.h 514 ConvolutionDescriptor& set_zero_padding(DimIndex dim, int64 value) { function in class:perftools::gputools::dnn::ConvolutionDescriptor
    [all...]

Completed in 2088 milliseconds