Searched refs:strides_attr (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_common.h | 250 Value input, Value filter, Value bias, ArrayAttr strides_attr,
|
| D | legalize_common.cc | 2231 DenseIntElementsAttr strides_attr; in convertStridedSliceOp() local
2233 if (!matchPattern(strides_value, m_Constant(&strides_attr))) { in convertStridedSliceOp()
2251 int32_t strides_size = strides_attr.getNumElements(); in convertStridedSliceOp()
3310 Value input, Value filter, Value bias, ArrayAttr strides_attr, in convertTFConv2DCommon() argument
3346 if (!strides_attr) { in convertTFConv2DCommon()
3350 int64_t stride_h = strides_attr[1].cast<IntegerAttr>().getInt(); in convertTFConv2DCommon()
3351 int64_t stride_w = strides_attr[2].cast<IntegerAttr>().getInt(); in convertTFConv2DCommon()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | prepare_tf.cc | 688 auto strides_attr = strided_slice_op.strides(); in matchAndRewrite() local
693 strides_attr.getType().dyn_cast<RankedTensorType>(); in matchAndRewrite()
707 !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/core/transforms/constant_folding/ |
| D | pass.cc | 1631 ElementsAttr strides_attr = in matchAndRewrite() local
1637 const int64_t num_strides_elements = strides_attr.getNumElements(); in matchAndRewrite()
1671 int s = strides_attr.getElementType().isInteger(32) in matchAndRewrite()
1672 ? strides_attr.getValues<int32_t>()[j] in matchAndRewrite()
1673 : strides_attr.getValues<int64_t>()[j]; in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_n_z.cc | 2055 DenseIntElementsAttr begin_attr, end_attr, strides_attr; in fold() local
2058 !matchPattern(strides(), m_Constant(&strides_attr)) || in fold()
2060 strides_attr.getNumElements() != 1) { in fold()
2076 int64_t strides_int = strides_attr.getValues<APInt>()[0].getSExtValue(); in fold()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 4923 auto strides_attr = GetI64ElementsAttr(op.strides()); in matchAndRewrite() local
4925 strides_attr.template getValues<int64_t>().begin(), in matchAndRewrite()
4926 strides_attr.template getValues<int64_t>().end()}; in matchAndRewrite()
5122 auto strides_attr = GetI64ElementsAttr(op.strides()); in matchAndRewrite() local
5124 strides_attr.template getValues<int64_t>().begin(), in matchAndRewrite()
5125 strides_attr.template getValues<int64_t>().end()}; in matchAndRewrite()
|