/external/tensorflow/tensorflow/compiler/xla/tests/ |
D | pad_test.cc | 369 constexpr int64 interior_padding = 0; in XLA_TEST_P() local 379 interior_padding); in XLA_TEST_P() 396 constexpr int64 interior_padding = 0; in XLA_TEST_P() local 407 interior_padding); in XLA_TEST_P() 424 int64 interior_padding[2] = {1, 2}; in XLA_TEST_P() local 435 interior_padding[dim]); in XLA_TEST_P()
|
/external/tensorflow/tensorflow/compiler/xla/ |
D | reference_util.h | 508 int64 interior_padding0 = padding.dimensions(0).interior_padding(); in PadArray2D() 515 int64 interior_padding1 = padding.dimensions(1).interior_padding(); in PadArray2D() 552 CHECK_LE(0, padding.dimensions(i).interior_padding()) in PadArray3D() 554 pad_interior[i] = padding.dimensions(i).interior_padding(); in PadArray3D() 609 CHECK_LE(0, padding.dimensions(i).interior_padding()) in PadArray4D() 611 pad_interior[i] = padding.dimensions(i).interior_padding(); in PadArray4D()
|
D | util.cc | 158 if (dim.interior_padding() != 0) { in HasInteriorPadding()
|
D | xla_data.proto | 105 int64 interior_padding = 3; field
|
/external/tensorflow/tensorflow/compiler/mlir/xla/ |
D | hlo_function_importer.cc | 412 llvm::SmallVector<int64_t, 4> interior_padding; in ImportInstructionImpl() local 415 interior_padding.reserve(padding_config.dimensions_size()); in ImportInstructionImpl() 420 interior_padding.push_back(dimension.interior_padding()); in ImportInstructionImpl() 427 Convert(interior_padding)) in ImportInstructionImpl()
|
D | mlir_hlo_to_hlo.cc | 821 auto interior_padding = ConvertDenseIntAttr(op.interior_padding()); in ExportXlaOp() local 826 dims->set_interior_padding(interior_padding[i]); in ExportXlaOp()
|
/external/tensorflow/tensorflow/compiler/xla/python/ |
D | xla_client.py | 454 self.interior_padding = 0 485 dimension.interior_padding = interior
|
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | dynamic_dimension_inference.cc | 371 if (padding_config.interior_padding() != 0) { in HandlePad() 378 HloInstruction* interior_padding = hlo->parent()->AddInstruction( in HandlePad() local 380 padding_config.interior_padding()))); in HandlePad() 392 dynamic_size_adjusted, interior_padding)); in HandlePad()
|
D | algebraic_simplifier.cc | 3335 if (padding_config.dimensions(i).interior_padding() > 0 && in HandlePad() 3365 } else if (padding_dimension.interior_padding()) { in HandlePad() 4164 const int64 interior = dim.interior_padding(); in HandleSlice() 4980 if (pad_dim.interior_padding() != 0) { in HandleReduceWindow() 4982 window_dim.set_base_dilation(1 + pad_dim.interior_padding()); in HandleReduceWindow() 5167 p.interior_padding() != 0) { in FoldConvInputPad() 5182 if (p.interior_padding() != 0 && in FoldConvInputPad() 5189 p.interior_padding() != 0)) { in FoldConvInputPad() 5195 if (p.interior_padding() != 0) { in FoldConvInputPad() 5197 w.set_base_dilation(1 + p.interior_padding()); in FoldConvInputPad() [all …]
|
D | shape_inference.cc | 554 return p.interior_padding() < 0; in InferPadShape() 571 p.interior_padding(); in InferPadShape()
|
D | elemental_ir_emitter.cc | 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()
|
D | hlo_instruction.cc | 3682 return dim.interior_padding() != 0; in PaddingConfigToString() 3689 has_interior_padding ? StrCat("_", dim.interior_padding()) : ""); in PaddingConfigToString()
|
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
D | strided_slice_op.cc | 546 (processing_shape.dim_size(i) - 1) * dims->interior_padding(); in Compile() 556 (processing_shape.dim_size(i) - 1) * dims->interior_padding(); in Compile()
|
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/ |
D | hlo_ops.cc | 1940 const auto& padding_interior = op.interior_padding(); in Verify() 1979 llvm::all_of(interior_padding().getIntValues(), is_zero)) in fold() 1986 llvm::all_of(interior_padding().getIntValues(), is_negative)) in fold() 2020 (interior_padding().getValue<int64_t>({uint64_t(i)}) + 1)) * in fold()
|
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/ |
D | legalize-tf.mlir | 1005 // CHECK-SAME: interior_padding = dense<0> : tensor<1xi64> 1016 // CHECK-SAME: interior_padding = dense<0> : tensor<2xi64> 1027 // CHECK-SAME: interior_padding = dense<0> : tensor<2xi64> 1880 …se<1> : tensor<1xi64>, edge_padding_low = dense<0> : tensor<1xi64>, interior_padding = dense<0> : … 4171 …tensor<3xi64>, edge_padding_low = dense<[0, 4, 0]> : tensor<3xi64>, interior_padding = dense<[0, 3… 4198 // CHECK-DAG-SAME: interior_padding = dense<0> : tensor<2xi64> 4225 // CHECK-DAG-SAME: interior_padding = dense<0> : tensor<1xi64> 4253 // CHECK-DAG-SAME: interior_padding = dense<0> : tensor<3xi64> 4280 // The edge_padding_low, edge_padding_high and interior_padding attributes of 4286 // CHECK-DAG-SAME: interior_padding = dense<[0, 0, 0, 0, 0, 1]> : tensor<6xi64> [all …]
|
D | legalize-tf-with-tf2xla.mlir | 137 // CHECK-SAME-DAG: interior_padding = dense<[1, 0]> : tensor<2xi64>
|
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/translate/ |
D | import.hlotxt | 575 …se<0> : tensor<1xi64>, edge_padding_low = dense<0> : tensor<1xi64>, interior_padding = dense<0> : … 584 …tensor<3xi64>, edge_padding_low = dense<[1, 3, 5]> : tensor<3xi64>, interior_padding = dense<0> : … 593 …se<0> : tensor<1xi64>, edge_padding_low = dense<0> : tensor<1xi64>, interior_padding = dense<2> : …
|
/external/tensorflow/tensorflow/compiler/mlir/xla/ir/ |
D | mlir_hlo_builder.cc | 551 internal.push_back(dimension.interior_padding()); in PadInternal()
|
/external/tensorflow/tensorflow/compiler/mlir/hlo/tests/ |
D | canonicalize.mlir | 1511 interior_padding = dense<0> : tensor<2xi64> 1524 interior_padding = dense<[0, 1]> : tensor<2xi64> 1535 …se<3> : tensor<1xi64>, edge_padding_low = dense<0> : tensor<1xi64>, interior_padding = dense<0> : …
|
/external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/IR/ |
D | lhlo_ops.td | 518 I64ElementsAttr:$interior_padding,
|
D | hlo_ops.td | 1191 I64ElementsAttr: $interior_padding
|
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/hlo_to_lhlo_with_xla/ |
D | hlo_text_to_lhlo_no_opt.hlotxt | 561 // CHECK-SAME: interior_padding = dense<1> : tensor<2xi64>}
|
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
D | legalize_tf.cc | 2413 DimVector interior_padding(num_dims, 0); in matchAndRewrite() local 2423 interior_padding[dim] = strides[dim] - 1; in matchAndRewrite() 2459 /*interior_padding=*/GetI64ElementsAttr(interior_padding, &rewriter)); in matchAndRewrite() 5436 llvm::SmallVector<int64_t, 4> interior_padding(rank, 0); in matchAndRewrite() local 5443 GetI64ElementsAttr(interior_padding, &rewriter)); in matchAndRewrite()
|
D | mhlo_to_lhlo_with_xla.cc | 1183 interior.push_back(dim.interior_padding()); in EmitPadOp()
|
/external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
D | spmd_partitioner.cc | 333 dim.interior_padding() == 0 && it->second.contains(i)) { in AddInstruction() 2638 dim->set_base_dilation(pd.interior_padding() + 1); in HandlePad()
|