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

  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
cudnn_convolution_rewriter.cc 86 auto kernel_spatial_dims = conv_dnums.kernel_spatial_dimensions(); local
197 for (int i = 0; i < kernel_spatial_dims.size(); ++i) {
198 backward_conv_dnums.add_output_spatial_dimensions(kernel_spatial_dims[i]);
227 const auto& kernel_spatial_dims = dnums.kernel_spatial_dimensions(); local
229 if (kernel_spatial_dims.size() != reverse_filter->dimensions().size() ||
230 !std::is_permutation(kernel_spatial_dims.begin(),
231 kernel_spatial_dims.end(),
239 for (int64 i = 0; i < kernel_spatial_dims.size(); ++i) {
384 AsInt64Slice(kernel_spatial_dims)));
  /external/tensorflow/tensorflow/compiler/tf2xla/kernels/
conv_ops.cc 409 std::vector<int64> kernel_spatial_dims(num_spatial_dims_);
420 kernel_spatial_dims[i] = i;
435 b->Rev(filter, kernel_spatial_dims);
  /external/tensorflow/tensorflow/compiler/xla/service/
shape_inference.cc     [all...]

Completed in 173 milliseconds