| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | pad_test.cc | 90 PaddingConfig padding_config; in XLA_TEST_P() local
105 PaddingConfig padding_config; in XLA_TEST_P() local
120 PaddingConfig padding_config; in XLA_TEST_P() local
188 PaddingConfig padding_config; in TEST_P() local
230 PaddingConfig padding_config; in XLA_TEST_P() local
327 PaddingConfig padding_config = MakeNoPaddingConfig(2); in XLA_TEST_P() local
349 PaddingConfig padding_config = MakeNoPaddingConfig(2); in XLA_TEST_P() local
373 PaddingConfig padding_config = MakeNoPaddingConfig(2); in XLA_TEST_P() local
400 PaddingConfig padding_config = MakeNoPaddingConfig(2); in XLA_TEST_P() local
428 PaddingConfig padding_config = MakeNoPaddingConfig(2); in XLA_TEST_P() local
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | pooling.cc | 54 PaddingConfig padding_config; in AvgPoolDivideByCountWithGeneralPadding() local
93 PaddingConfig padding_config; in MakeSpatialPaddingConfig() local
161 auto padding_config = MakeSpatialPaddingConfig(padding, num_spatial_dims, in AvgPool() local
239 PaddingConfig padding_config = MakeNoPaddingConfig(num_dims); in AvgPoolGrad() local
|
| D | matrix.cc | 190 PaddingConfig padding_config; in SetMatrixDiagonal() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | xla_pad_op.cc | 80 xla::PaddingConfig padding_config; in Compile() local
|
| D | spacetobatch_op.cc | 53 xla::PaddingConfig padding_config; in SpaceToBatch() local
|
| D | matrix_diag_ops.cc | 126 xla::PaddingConfig padding_config; in SetMatrixDiag() local
399 xla::PaddingConfig padding_config = in Compile() local
|
| D | strided_slice_op.cc | 254 xla::PaddingConfig padding_config; in Compile() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_conv_padding_legalization.cc | 62 PaddingConfig padding_config = in MaybePaddedAndSlicedInput() local
131 PaddingConfig padding_config; in MaybePaddedKernel() local
|
| D | cublas_gemm_pad_for_tensor_cores.cc | 68 PaddingConfig padding_config; in PadForTensorCores() local
|
| D | tree_reduction_rewriter.cc | 111 PaddingConfig padding_config = MakeNoPaddingConfig(input_shape.rank()); in RewriteReduction() local
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | util.cc | 170 PaddingConfig padding_config; in MakeNoPaddingConfig() local
182 PaddingConfig padding_config; in MakeEdgePaddingConfig() local
|
| D | window_util.cc | 162 bool HasSymmetricPadding(const PaddingConfig& padding_config) { in HasSymmetricPadding()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_creation_utils.cc | 60 const PaddingConfig& padding_config) { in MakePadHlo()
478 PaddingConfig padding_config; in PadVectorWithZeros() local
|
| D | hlo_verifier_test.cc | 358 PaddingConfig padding_config; in TEST_F() local
382 PaddingConfig padding_config; in TEST_F() local
|
| D | triangular_solve_expander.cc | 67 PaddingConfig padding_config = in DiagonalBlocks() local
|
| D | hlo.proto | 112 xla.PaddingConfig padding_config = 21; field
|
| D | hlo_instructions.h | 1346 const PaddingConfig& padding_config() const { return padding_config_; } in padding_config() function
|
| D | algebraic_simplifier.cc | 906 PaddingConfig padding_config; in HandleConcatenate() local
2687 PaddingConfig padding_config = pad->padding_config(); in HandlePad() local
3212 auto padding_config = pad->padding_config(); in TrySimplifyScalarSlice() local
|
| D | hlo_evaluator_test.cc | 636 PaddingConfig padding_config; in CreatePaddingConfig() local
658 auto padding_config = CreatePaddingConfig({{{1, 0, 2}}, {{0, 2, 1}}}); in TEST_F() local
767 PaddingConfig padding_config = MakeNoPaddingConfig(2); in TEST_P() local
|
| D | dynamic_dimension_inference.cc | 239 const PaddingConfig_PaddingConfigDimension& padding_config = in HandlePad() local
|
| D | hlo_instruction.cc | 1275 const PaddingConfig& padding_config) { in CreatePad()
3846 const PaddingConfig& HloInstruction::padding_config() const { in padding_config() function in xla::HloInstruction
|
| D | memory_space_assignment_test.cc | 1342 PaddingConfig padding_config = MakeEdgePaddingConfig({{0, 0}, {0, 1}}); in TEST_P() local
1405 PaddingConfig padding_config = MakeEdgePaddingConfig({{0, 0}, {0, 1}}); in TEST_P() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | hlo_function_importer.cc | 327 const auto& padding_config = instruction->padding_config(); in ImportInstruction() local
|
| D | mlir_hlo_to_hlo.cc | 600 xla::PaddingConfig padding_config; in ExportXlaOp() local
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder_test.cc | 661 PaddingConfig padding_config; in TEST_F() local
|