Searched refs:dilations_attr (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | dilated_conv.h | 198 llvm::Optional<ArrayAttr> dilations_attr = ExtractDilationsAttrFromBlockShape( in matchAndRewrite() local
200 if (!dilations_attr.hasValue()) return failure(); in matchAndRewrite()
256 op->setAttr("dilations", dilations_attr.getValue()); in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_common.h | 237 ArrayAttr dilations_attr, ArrayAttr explicit_padding_attr,
|
| D | legalize_common.cc | 2783 ArrayAttr dilations_attr, ArrayAttr explicit_padding_attr, in convertTFConv2DCommon() argument
2826 if (!dilations_attr) { in convertTFConv2DCommon()
2830 int64_t dilation_h = dilations_attr[1].cast<IntegerAttr>().getInt(); in convertTFConv2DCommon()
2831 int64_t dilation_w = dilations_attr[2].cast<IntegerAttr>().getInt(); in convertTFConv2DCommon()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | generic_layout_optimizer_transposer_test.cc | 1078 auto* dilations_attr = conv2d_bf_node->GetAttr("dilations"); in TEST_F() local
1079 ASSERT_NE(dilations_attr, nullptr); in TEST_F()
1080 ASSERT_EQ(dilations_attr->list().i_size(), 4); in TEST_F()
1081 EXPECT_EQ(dilations_attr->list().i(0), 1); in TEST_F()
1082 EXPECT_EQ(dilations_attr->list().i(1), 1); in TEST_F()
1083 EXPECT_EQ(dilations_attr->list().i(2), kDilation); in TEST_F()
1084 EXPECT_EQ(dilations_attr->list().i(3), kDilation); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 4128 auto dilations_attr = GetI64ElementsAttr(op.dilations()); in matchAndRewrite() local
4130 dilations_attr.template getValues<int64_t>().begin(), in matchAndRewrite()
4131 dilations_attr.template getValues<int64_t>().end()}; in matchAndRewrite()
4311 auto dilations_attr = GetI64ElementsAttr(op.dilations()); in matchAndRewrite() local
4313 dilations_attr.template getValues<int64_t>().begin(), in matchAndRewrite()
4314 dilations_attr.template getValues<int64_t>().end()}; in matchAndRewrite()
|