Home
last modified time | relevance | path

Searched refs:pad_dim (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_pad_for_convolutions.cc234 auto pad_dim = [](Shape* s, int64 dim) { in TryResolvePaddedShapesForTensorCore() local
237 pad_dim(new_input_shape, dnums.input_feature_dimension()); in TryResolvePaddedShapesForTensorCore()
238 pad_dim(new_filter_shape, dnums.kernel_input_feature_dimension()); in TryResolvePaddedShapesForTensorCore()
239 pad_dim(new_filter_shape, dnums.kernel_output_feature_dimension()); in TryResolvePaddedShapesForTensorCore()
240 pad_dim(new_output_shape, dnums.output_feature_dimension()); in TryResolvePaddedShapesForTensorCore()
310 auto pad_dim = [](Shape* s, int64 dim) { in TryResolvePaddedShapesForIntegerConvolution() local
317 pad_dim(&new_input_shapes[0], in TryResolvePaddedShapesForIntegerConvolution()
319 pad_dim(&new_input_shapes[1], in TryResolvePaddedShapesForIntegerConvolution()
322 pad_dim( in TryResolvePaddedShapesForIntegerConvolution()
326 pad_dim(&new_result_shape, in TryResolvePaddedShapesForIntegerConvolution()
[all …]
Dcublas_gemm_pad_for_tensor_cores.cc42 auto pad_dim = [](Shape& s, int64 dim) { in PadForTensorCores() local
46 auto pad_matrix_dims = [&pad_dim](Shape s) { in PadForTensorCores()
50 pad_dim(s, s.rank() - 2); in PadForTensorCores()
51 pad_dim(s, s.rank() - 1); in PadForTensorCores()
/external/tensorflow/tensorflow/compiler/xla/service/
Delemental_ir_emitter.cc2119 const auto& pad_dim = hlo->padding_config().dimensions(i); in EmitElementalPad() local
2121 Sub(multi_index[i], index_typed_const(pad_dim.edge_padding_low())); in EmitElementalPad()
2128 index_typed_const(pad_dim.interior_padding() + 1))), in EmitElementalPad()
2131 SDiv(multi_index[i], index_typed_const(pad_dim.interior_padding() + 1)); in EmitElementalPad()
Dalgebraic_simplifier.cc4974 const auto& pad_dim = pad_config.dimensions(i); in HandleReduceWindow() local
4977 pad_dim.edge_padding_low()); in HandleReduceWindow()
4979 pad_dim.edge_padding_high()); in HandleReduceWindow()
4980 if (pad_dim.interior_padding() != 0) { in HandleReduceWindow()
4982 window_dim.set_base_dilation(1 + pad_dim.interior_padding()); in HandleReduceWindow()