Home
last modified time | relevance | path

Searched refs:edge_padding_high (Results 1 – 25 of 29) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/xla/
Dreference_util.h506 int64 high_padding0 = padding.dimensions(0).edge_padding_high(); in PadArray2D()
513 int64 high_padding1 = padding.dimensions(1).edge_padding_high(); in PadArray2D()
549 pad_high[i] = padding.dimensions(i).edge_padding_high(); in PadArray3D()
608 pad_high[i] = padding.dimensions(i).edge_padding_high(); in PadArray4D()
Dwindow_util.cc166 dim.edge_padding_high(); in HasSymmetricPadding()
Dxla_data.proto102 int64 edge_padding_high = 2; field
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dhlo_function_importer.cc411 llvm::SmallVector<int64_t, 4> edge_padding_high; in ImportInstructionImpl() local
414 edge_padding_high.reserve(padding_config.dimensions_size()); in ImportInstructionImpl()
419 edge_padding_high.push_back(dimension.edge_padding_high()); in ImportInstructionImpl()
426 Convert(edge_padding_high), in ImportInstructionImpl()
Dmlir_hlo_to_hlo.cc820 auto edge_padding_high = ConvertDenseIntAttr(op.edge_padding_high()); in ExportXlaOp() local
825 dims->set_edge_padding_high(edge_padding_high[i]); in ExportXlaOp()
/external/tensorflow/tensorflow/compiler/xla/python/
Dxla_client.py453 self.edge_padding_high = 0
484 dimension.edge_padding_high = hi
Dxla_client_test.py1062 dimension.edge_padding_high = hi
/external/tensorflow/tensorflow/compiler/xla/service/
Dalgebraic_simplifier.cc3358 padding_dimension.edge_padding_high() < 0) { in HandlePad()
3362 padding_dimension.edge_padding_high() != 0) { in HandlePad()
3471 if (padding_dimension->edge_padding_high() < 0) { in HandlePad()
3498 if (padding_dimension.edge_padding_high() < 0) { in HandlePad()
3499 end += padding_dimension.edge_padding_high(); in HandlePad()
4163 const int64 high = dim.edge_padding_high(); in HandleSlice()
4913 pad_dimension.edge_padding_high() != 0) && in HandleReduceWindow()
4925 pad_dimension.edge_padding_high() != 0); in HandleReduceWindow()
4979 pad_dim.edge_padding_high()); in HandleReduceWindow()
5166 if (p.edge_padding_low() != 0 || p.edge_padding_high() != 0 || in FoldConvInputPad()
[all …]
Ddynamic_dimension_inference.cc401 padding_config.edge_padding_high()))); in HandlePad()
Dalgebraic_simplifier_test.cc3301 padding_dimension.edge_padding_high() < 0) { in TEST_F()
3353 padding_dimension.edge_padding_high() < 0) { in TEST_F()
7306 EXPECT_EQ(pad->padding_config().dimensions(0).edge_padding_high(), 0); in TEST_F()
Dshape_inference.cc569 p.edge_padding_high() + in InferPadShape()
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/
Dlegalize-tf.mlir1003 // CHECK-SAME: edge_padding_high = dense<2> : tensor<1xi64>,
1014 // CHECK-SAME: edge_padding_high = dense<[2, 4]> : tensor<2xi64>,
1025 // CHECK-SAME: edge_padding_high = dense<[2, 4]> : tensor<2xi64>,
1880 …// CHECK: %[[PADDED:.*]] = "mhlo.pad"(%arg0, %2) {edge_padding_high = dense<1> : tensor<1xi64>, ed…
4171 …// CHECK: [[PAD:%.*]] = "mhlo.pad"([[REVERSE]], [[ZERO]]) {edge_padding_high = dense<[0, 63, 2]> :…
4197 // CHECK-DAG-SAME: edge_padding_high = dense<[1, 0]> : tensor<2xi64>
4224 // CHECK-DAG-SAME: edge_padding_high = dense<4> : tensor<1xi64>
4252 // CHECK-DAG-SAME: edge_padding_high = dense<0> : tensor<3xi64>
4280 // The edge_padding_low, edge_padding_high and interior_padding attributes of
4285 // CHECK-DAG-SAME: edge_padding_high = dense<[0, 0, 0, 0, 22, 59]> : tensor<6xi64>
[all …]
Dlegalize-tf-with-tf2xla.mlir135 // CHECK-SAME-DAG: edge_padding_high = dense<[1, 2]> : tensor<2xi64>
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/translate/
Dimport.hlotxt575 …// CHECK-NEXT: "mhlo.pad"(%arg0, %arg1) {edge_padding_high = dense<0> : tensor<1xi64>, edge_paddi…
584 …// CHECK-NEXT: "mhlo.pad"(%arg0, %arg1) {edge_padding_high = dense<[2, 4, 6]> : tensor<3xi64>, ed…
593 …// CHECK-NEXT: "mhlo.pad"(%arg0, %arg1) {edge_padding_high = dense<0> : tensor<1xi64>, edge_paddi…
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.cc555 dims->edge_padding_high() + processing_shape.dim_size(i) + in Compile()
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/
Dhlo_ops.cc1933 const auto& padding_high = op.edge_padding_high(); in Verify()
1978 llvm::all_of(edge_padding_high().getIntValues(), is_zero) && in fold()
1985 llvm::all_of(edge_padding_high().getIntValues(), is_negative) && in fold()
/external/tensorflow/tensorflow/compiler/mlir/xla/ir/
Dmlir_hlo_builder.cc550 high.push_back(dimension.edge_padding_high()); in PadInternal()
/external/tensorflow/tensorflow/compiler/mlir/hlo/tests/
Dcanonicalize.mlir1510 edge_padding_high = dense<0> : tensor<2xi64>,
1523 edge_padding_high = dense<[1, 2]> : 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.td517 I64ElementsAttr:$edge_padding_high,
Dhlo_ops.td1190 I64ElementsAttr: $edge_padding_high,
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/hlo_to_lhlo_with_xla/
Dhlo_text_to_lhlo_no_opt.hlotxt559 // CHECK-SAME: edge_padding_high = dense<[4, 5]> : tensor<2xi64>,
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dlegalize_hlo.cc1176 pad_op.edge_padding_high().getValues<APInt>())) { in ConvertPadOp()
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dmhlo_to_lhlo_with_xla.cc1182 high.push_back(dim.edge_padding_high()); 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()
2637 dim->set_padding_high(pd.edge_padding_high()); in HandlePad()

12