HomeSort by relevance Sort by last modified time
    Searched defs:out_depths (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
cudnn_convolution_algorithm_picker.cc 109 int64 out_depths = output_shape.dimensions(dnums.output_feature_dimension()); local
112 std::max(in_depths, out_depths) * in_cols * in_rows *
  /external/tensorflow/tensorflow/core/kernels/
conv_ops.cc 567 const int64 out_depths = GetTensorDim(*output, data_format, 'C'); local
641 .set_feature_map_count(out_depths)
671 out_cols, out_depths),
695 out_depths, // out_depths
    [all...]

Completed in 3891 milliseconds