Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/tests/
Dpad_test.cc325 PaddingConfig padding_config = MakeNoPaddingConfig(2); in XLA_TEST_P()
347 PaddingConfig padding_config = MakeNoPaddingConfig(2); in XLA_TEST_P()
371 PaddingConfig padding_config = MakeNoPaddingConfig(2); in XLA_TEST_P()
398 PaddingConfig padding_config = MakeNoPaddingConfig(2); in XLA_TEST_P()
426 PaddingConfig padding_config = MakeNoPaddingConfig(2); in XLA_TEST_P()
454 PaddingConfig padding_config = MakeNoPaddingConfig(3); in XLA_TEST_P()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_conv_padding_legalization.cc64 MakeNoPaddingConfig(input->shape().dimensions_size()); in MaybePaddedAndSlicedInput()
225 MakeNoPaddingConfig(input->shape().rank()); in CanonicalizeBackwardFilterConvolution()
Dtree_reduction_rewriter.cc137 MakeNoPaddingConfig(input_shape_dims.size()); in RewriteReduction()
Dcudnn_pad_for_convolutions.cc47 PaddingConfig pad_config = MakeNoPaddingConfig(shape.rank()); in PadInstruction()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dmatrix_diag_ops.cc130 padding_config = xla::MakeNoPaddingConfig(input_rank - 1); in SetMatrixDiag()
404 xla::MakeNoPaddingConfig(input_rank - 1); in Compile()
Dimage_resize_ops.cc463 xla::PaddingConfig padding = xla::MakeNoPaddingConfig(4); in ResizeUsingDilationAndConvolutionGradOp()
/external/tensorflow/tensorflow/compiler/xla/service/
Dtriangular_solve_expander.cc92 PaddingConfig config = MakeNoPaddingConfig(ndims); in DiagonalBlocks()
100 config = MakeNoPaddingConfig(2); in DiagonalBlocks()
Dspace_to_batch_converter.cc565 MakeNoPaddingConfig(first_slice->shape().dimensions_size()); in HaloDuplicateWithSlice()
587 MakeNoPaddingConfig(halo_region->shape().dimensions_size()); in HaloDuplicateWithSlice()
839 PaddingConfig padding_config = MakeNoPaddingConfig( in ChangeSpatialSizeOnSpaceToBatchedShape()
1831 MakeNoPaddingConfig(reshape->shape().dimensions_size()); in Propagate()
2198 MakeNoPaddingConfig(final_selection->shape().dimensions_size()); in Propagate()
2912 MakeNoPaddingConfig(activations->shape().dimensions_size()); in PadAndSplitSpace()
Deigh_expander.cc484 PaddingConfig config = MakeNoPaddingConfig(num_dims); in BuildEigh()
Dhlo_evaluator_test.cc806 PaddingConfig padding_config = MakeNoPaddingConfig(2); in TEST_P()
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dpooling.cc242 PaddingConfig padding_config = MakeNoPaddingConfig(num_dims); in AvgPoolGrad()
/external/tensorflow/tensorflow/compiler/xla/
Dutil.cc193 PaddingConfig MakeNoPaddingConfig(int64_t rank) { in MakeNoPaddingConfig() function
Dutil.h353 PaddingConfig MakeNoPaddingConfig(int64_t rank);
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner.cc4769 MakeNoPaddingConfig(hlo->shape().rank()); in PreprocessHlos()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc1164 PaddingConfig padding_config = MakeNoPaddingConfig(shape->rank()); in PadInDim()