Searched refs:strides_attr (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_common.h | 236 Value input, Value filter, Value bias, ArrayAttr strides_attr,
|
| D | legalize_common.cc | 2782 Value input, Value filter, Value bias, ArrayAttr strides_attr, in convertTFConv2DCommon() argument
2816 if (!strides_attr) { in convertTFConv2DCommon()
2820 int64_t stride_h = strides_attr[1].cast<IntegerAttr>().getInt(); in convertTFConv2DCommon()
2821 int64_t stride_w = strides_attr[2].cast<IntegerAttr>().getInt(); in convertTFConv2DCommon()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | prepare_tf.cc | 767 auto strides_attr = strided_slice_op.strides(); in matchAndRewrite() local
772 strides_attr.getType().dyn_cast<RankedTensorType>(); in matchAndRewrite()
786 !matchPattern(strides_attr, m_Constant(&strides_elem_attr))) { in matchAndRewrite()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | generic_layout_optimizer_transposer_test.cc | 638 const auto& strides_attr = conv2d->GetAttr("strides"); in TEST_F() local
639 ASSERT_NE(strides_attr, nullptr); in TEST_F()
640 EXPECT_EQ(strides_attr->list().DebugString(), in TEST_F()
858 const auto* strides_attr = conv2d_node->GetAttr("strides"); in TEST_F() local
859 ASSERT_NE(strides_attr, nullptr); in TEST_F()
860 ASSERT_EQ(strides_attr->list().i_size(), 4); in TEST_F()
861 EXPECT_EQ(strides_attr->list().i(0), 1); in TEST_F()
862 EXPECT_EQ(strides_attr->list().i(1), 1); in TEST_F()
863 EXPECT_EQ(strides_attr->list().i(2), kStride1); in TEST_F()
864 EXPECT_EQ(strides_attr->list().i(3), kStride2); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_n_z.cc | 1915 DenseIntElementsAttr begin_attr, end_attr, strides_attr; in fold() local
1918 !matchPattern(strides(), m_Constant(&strides_attr)) || in fold()
1920 strides_attr.getNumElements() != 1) { in fold()
1936 int64_t strides_int = strides_attr.getValue<APInt>(0).getSExtValue(); in fold()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 4132 auto strides_attr = GetI64ElementsAttr(op.strides()); in matchAndRewrite() local
4134 strides_attr.template getValues<int64_t>().begin(), in matchAndRewrite()
4135 strides_attr.template getValues<int64_t>().end()}; in matchAndRewrite()
4315 auto strides_attr = GetI64ElementsAttr(op.strides()); in matchAndRewrite() local
4317 strides_attr.template getValues<int64_t>().begin(), in matchAndRewrite()
4318 strides_attr.template getValues<int64_t>().end()}; in matchAndRewrite()
|