Home
last modified time | relevance | path

Searched refs:edge_padding_low (Results 1 – 25 of 32) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/xla/
Dreference_util.h507 int64 low_padding0 = padding.dimensions(0).edge_padding_low(); in PadArray2D()
514 int64 low_padding1 = padding.dimensions(1).edge_padding_low(); in PadArray2D()
548 pad_low[i] = padding.dimensions(i).edge_padding_low(); in PadArray3D()
607 pad_low[i] = padding.dimensions(i).edge_padding_low(); in PadArray4D()
Dwindow_util.cc165 return dim.edge_padding_low() == in HasSymmetricPadding()
Dxla_data.proto98 int64 edge_padding_low = 1; field
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dhlo_function_importer.cc410 llvm::SmallVector<int64_t, 4> edge_padding_low; in ImportInstructionImpl() local
413 edge_padding_low.reserve(padding_config.dimensions_size()); in ImportInstructionImpl()
418 edge_padding_low.push_back(dimension.edge_padding_low()); in ImportInstructionImpl()
425 operands[1], Convert(edge_padding_low), in ImportInstructionImpl()
Dmlir_hlo_to_hlo.cc819 auto edge_padding_low = ConvertDenseIntAttr(op.edge_padding_low()); in ExportXlaOp() local
822 for (xla::int64 i = 0, end = edge_padding_low.size(); i < end; ++i) { in ExportXlaOp()
824 dims->set_edge_padding_low(edge_padding_low[i]); in ExportXlaOp()
/external/tensorflow/tensorflow/compiler/xla/python/
Dxla_client.py452 self.edge_padding_low = 0
483 dimension.edge_padding_low = lo
/external/tensorflow/tensorflow/compiler/xla/service/
Dalgebraic_simplifier.cc3357 if (padding_dimension.edge_padding_low() < 0 || in HandlePad()
3361 if (padding_dimension.edge_padding_low() != 0 || in HandlePad()
3468 if (padding_dimension->edge_padding_low() < 0) { in HandlePad()
3494 if (padding_dimension.edge_padding_low() < 0) { in HandlePad()
3495 start = -1 * padding_dimension.edge_padding_low(); in HandlePad()
4162 const int64 low = dim.edge_padding_low(); in HandleSlice()
4912 if ((pad_dimension.edge_padding_low() != 0 || in HandleReduceWindow()
4924 bool dimension_has_padding = (pad_dimension.edge_padding_low() != 0 || in HandleReduceWindow()
4927 window_dimension.size() < pad_dimension.edge_padding_low() + 1) { in HandleReduceWindow()
4977 pad_dim.edge_padding_low()); in HandleReduceWindow()
[all …]
Ddynamic_dimension_inference.cc400 padding_config.edge_padding_low() + in HandlePad()
Ddynamic_padder.cc783 LiteralUtil::CreateR0<int32>(padding_dim->edge_padding_low()))), in RewriteInputWithDynamicPadding()
Dalgebraic_simplifier_test.cc3300 if (padding_dimension.edge_padding_low() < 0 || in TEST_F()
3352 if (padding_dimension.edge_padding_low() < 0 || in TEST_F()
7305 EXPECT_EQ(pad->padding_config().dimensions(0).edge_padding_low(), 1); in TEST_F()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dlegalize_hlo.cc1175 for (auto p : llvm::zip(pad_op.edge_padding_low().getValues<APInt>(), in ConvertPadOp()
1180 auto attr_type = RankedTensorType::get({pad_op.edge_padding_low().size(), 2}, in ConvertPadOp()
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/
Dhlo_ops.cc1926 const auto& padding_low = op.edge_padding_low(); in Verify()
1977 if (llvm::all_of(edge_padding_low().getIntValues(), is_zero) && in fold()
1984 if (llvm::all_of(edge_padding_low().getIntValues(), is_negative) && in fold()
2018 (edge_padding_low().getValue<int64_t>({uint64_t(i)}) + in fold()
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/
Dlegalize-tf.mlir1004 // CHECK-SAME: edge_padding_low = dense<1> : tensor<1xi64>,
1015 // CHECK-SAME: edge_padding_low = dense<[1, 3]> : tensor<2xi64>,
1026 // CHECK-SAME: edge_padding_low = dense<[1, 3]> : tensor<2xi64>,
1880 …mhlo.pad"(%arg0, %2) {edge_padding_high = dense<1> : tensor<1xi64>, edge_padding_low = dense<0> : …
4171 …, [[ZERO]]) {edge_padding_high = dense<[0, 63, 2]> : tensor<3xi64>, edge_padding_low = dense<[0, 4…
4196 // CHECK-DAG-SAME: edge_padding_low = dense<[2, 0]> : tensor<2xi64>
4223 // CHECK-DAG-SAME: edge_padding_low = dense<2> : tensor<1xi64>
4251 // CHECK-DAG-SAME: edge_padding_low = dense<[2, 0, 0]> : tensor<3xi64>
4280 // The edge_padding_low, edge_padding_high and interior_padding attributes of
4284 // CHECK-DAG-SAME: edge_padding_low = dense<[1, 0, 0, 8, 0, 2]> : tensor<6xi64>
[all …]
Dlegalize-tf-with-tf2xla.mlir136 // CHECK-SAME-DAG: edge_padding_low = dense<[2, 1]> : tensor<2xi64>
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/translate/
Dimport.hlotxt575 …o.pad"(%arg0, %arg1) {edge_padding_high = dense<0> : tensor<1xi64>, edge_padding_low = dense<0> : …
584 …arg0, %arg1) {edge_padding_high = dense<[2, 4, 6]> : tensor<3xi64>, edge_padding_low = dense<[1, 3…
593 …o.pad"(%arg0, %arg1) {edge_padding_high = dense<0> : tensor<1xi64>, edge_padding_low = dense<0> : …
Dexport.mlir592 …%0 = "mhlo.pad"(%arg, %pad) {edge_padding_high = dense<[4,5]> : tensor<2xi64>, edge_padding_low = …
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dstrided_slice_op.cc545 dims->edge_padding_low() + processing_shape.dim_size(i) + in Compile()
/external/tensorflow/tensorflow/compiler/mlir/xla/ir/
Dmlir_hlo_builder.cc549 low.push_back(dimension.edge_padding_low()); in PadInternal()
/external/tensorflow/tensorflow/compiler/mlir/hlo/tests/
Dcanonicalize.mlir1509 edge_padding_low = dense<0> : tensor<2xi64>,
1522 edge_padding_low = dense<[1, 0]> : tensor<2xi64>,
1535 …%2 = "mhlo.pad"(%0, %1) {edge_padding_high = dense<3> : tensor<1xi64>, edge_padding_low = dense<0>…
/external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/IR/
Dlhlo_ops.td516 I64ElementsAttr:$edge_padding_low,
Dhlo_ops.td1189 I64ElementsAttr: $edge_padding_low,
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/hlo_to_lhlo_with_xla/
Dhlo_text_to_lhlo_no_opt.hlotxt560 // CHECK-SAME: edge_padding_low = dense<[2, 3]> : tensor<2xi64>,
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc2008 if (padding_dimension.edge_padding_low() < 0 || in HandlePad()
2049 offset, b_.getInt64(padding_config.dimensions(i).edge_padding_low())); in HandlePad()
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dmhlo_to_lhlo_with_xla.cc1181 low.push_back(dim.edge_padding_low()); in EmitPadOp()
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner.cc332 if (dim.edge_padding_low() == 0 && dim.edge_padding_high() == 0 && in AddInstruction()
2636 dim->set_padding_low(pd.edge_padding_low()); in HandlePad()

12