Home
last modified time | relevance | path

Searched refs:MakeNoPaddingConfig (Results 1 – 14 of 14) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
Dpad_test.cc327 PaddingConfig padding_config = MakeNoPaddingConfig(2); in XLA_TEST_P()
349 PaddingConfig padding_config = MakeNoPaddingConfig(2); in XLA_TEST_P()
373 PaddingConfig padding_config = MakeNoPaddingConfig(2); in XLA_TEST_P()
400 PaddingConfig padding_config = MakeNoPaddingConfig(2); in XLA_TEST_P()
428 PaddingConfig padding_config = MakeNoPaddingConfig(2); in XLA_TEST_P()
456 PaddingConfig padding_config = MakeNoPaddingConfig(3); in XLA_TEST_P()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_conv_padding_legalization.cc63 MakeNoPaddingConfig(input->shape().dimensions_size()); in MaybePaddedAndSlicedInput()
222 MakeNoPaddingConfig(input->shape().rank()); in CanonicalizeBackwardFilterConvolution()
Dtree_reduction_rewriter.cc125 PaddingConfig padding_config = MakeNoPaddingConfig(input_shape.rank()); in RewriteReduction()
Dcudnn_pad_for_convolutions.cc53 PaddingConfig pad_config = MakeNoPaddingConfig(shape.rank()); in PadInstruction()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dmatrix_diag_ops.cc127 padding_config = xla::MakeNoPaddingConfig(input_rank - 1); in SetMatrixDiag()
401 xla::MakeNoPaddingConfig(input_rank - 1); in Compile()
Dimage_resize_ops.cc457 xla::PaddingConfig padding = xla::MakeNoPaddingConfig(4); in ResizeUsingDilationAndConvolutionGradOp()
/external/tensorflow/tensorflow/compiler/xla/service/
Dtriangular_solve_expander.cc98 PaddingConfig config = MakeNoPaddingConfig(ndims); in DiagonalBlocks()
106 config = MakeNoPaddingConfig(2); in DiagonalBlocks()
Dspace_to_batch_converter.cc400 MakeNoPaddingConfig(first_slice->shape().dimensions_size()); in HaloDuplicateWithSlice()
422 MakeNoPaddingConfig(halo_region->shape().dimensions_size()); in HaloDuplicateWithSlice()
592 MakeNoPaddingConfig(reshaped_activations->shape().dimensions_size()); in IncreaseSpatialSizeOnSpaceToBatchedShape()
1294 MakeNoPaddingConfig(reshape->shape().dimensions_size()); in Propagate()
1561 MakeNoPaddingConfig(final_selection->shape().dimensions_size()); in Propagate()
2048 MakeNoPaddingConfig(activations->shape().dimensions_size()); in SplitSpaceHelper()
Dhlo_evaluator_test.cc768 PaddingConfig padding_config = MakeNoPaddingConfig(2); in TEST_P()
Dalgebraic_simplifier_test.cc4349 PaddingConfig padding = MakeNoPaddingConfig(4); in TEST_F()
4431 PaddingConfig padding = MakeNoPaddingConfig(4); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/
Dutil.cc133 PaddingConfig MakeNoPaddingConfig(int64 rank) { in MakeNoPaddingConfig() function
Dutil.h389 PaddingConfig MakeNoPaddingConfig(int64 rank);
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dpooling.cc242 PaddingConfig padding_config = MakeNoPaddingConfig(num_dims); in AvgPoolGrad()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc1097 PaddingConfig padding_config = MakeNoPaddingConfig(shape->rank()); in PadInDim()