Home
last modified time | relevance | path

Searched refs:interior_padding (Results 1 – 25 of 31) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/xla/tests/
Dpad_test.cc369 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/
Dreference_util.h508 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()
Dutil.cc158 if (dim.interior_padding() != 0) { in HasInteriorPadding()
Dxla_data.proto105 int64 interior_padding = 3; field
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dhlo_function_importer.cc412 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()
Dmlir_hlo_to_hlo.cc821 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/
Dxla_client.py454 self.interior_padding = 0
485 dimension.interior_padding = interior
/external/tensorflow/tensorflow/compiler/xla/service/
Ddynamic_dimension_inference.cc371 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()
Dalgebraic_simplifier.cc3335 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 …]
Dshape_inference.cc554 return p.interior_padding() < 0; in InferPadShape()
571 p.interior_padding(); in InferPadShape()
Delemental_ir_emitter.cc2128 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()
Dhlo_instruction.cc3682 return dim.interior_padding() != 0; in PaddingConfigToString()
3689 has_interior_padding ? StrCat("_", dim.interior_padding()) : ""); in PaddingConfigToString()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dstrided_slice_op.cc546 (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/
Dhlo_ops.cc1940 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/
Dlegalize-tf.mlir1005 // 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 …]
Dlegalize-tf-with-tf2xla.mlir137 // CHECK-SAME-DAG: interior_padding = dense<[1, 0]> : tensor<2xi64>
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/translate/
Dimport.hlotxt575 …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/
Dmlir_hlo_builder.cc551 internal.push_back(dimension.interior_padding()); in PadInternal()
/external/tensorflow/tensorflow/compiler/mlir/hlo/tests/
Dcanonicalize.mlir1511 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/
Dlhlo_ops.td518 I64ElementsAttr:$interior_padding,
Dhlo_ops.td1191 I64ElementsAttr: $interior_padding
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/hlo_to_lhlo_with_xla/
Dhlo_text_to_lhlo_no_opt.hlotxt561 // CHECK-SAME: interior_padding = dense<1> : tensor<2xi64>}
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc2413 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()
Dmhlo_to_lhlo_with_xla.cc1183 interior.push_back(dim.interior_padding()); in EmitPadOp()
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner.cc333 dim.interior_padding() == 0 && it->second.contains(i)) { in AddInstruction()
2638 dim->set_base_dilation(pd.interior_padding() + 1); in HandlePad()

12