Home
last modified time | relevance | path

Searched refs:Reshape (Results 1 – 25 of 251) sorted by relevance

1234567891011

/external/tensorflow/tensorflow/compiler/xla/service/
Dbatch_dot_simplification_test.cc47 op::Reshape(op::Dot( in TEST_F()
48 op::Reshape(op::Parameter(0)), op::Reshape(op::Parameter(1)), in TEST_F()
71 op::Reshape(op::Dot( in TEST_F()
72 op::Reshape(op::Parameter(0)), op::Reshape(op::Parameter(1)), in TEST_F()
95 op::Reshape(op::Dot( in TEST_F()
96 op::Reshape(op::Parameter(0)), op::Reshape(op::Parameter(1)), in TEST_F()
119 op::Reshape(op::Dot( in TEST_F()
120 op::Reshape(op::Parameter(0)), op::Reshape(op::Parameter(1)), in TEST_F()
143 op::Reshape(op::Dot( in TEST_F()
144 op::Reshape(op::Parameter(0)), op::Reshape(op::Parameter(1)), in TEST_F()
[all …]
Dreshape_mover_test.cc57 op::Add(op::Reshape(param0), op::Reshape(param1))); in TEST_F()
62 op::Add(op::Reshape(param0), op::Reshape(param1))); in TEST_F()
100 op::Add(op::Reshape(rng0), const1));
105 op::Add(op::Reshape(rng0), const1));
126 op::Add(op::Reshape(param0), op::Reshape(param1))); in TEST_F()
132 op::Add(op::Reshape(op::Parameter()), op::Reshape(op::Parameter()))); in TEST_F()
153 op::Add(op::Reshape(param0), op::Reshape(param1))); in TEST_F()
157 op::Reshape(op::Add(param0, param1))); in TEST_F()
212 op::Reshape(op::Select(op::Reshape(const0), param1, param2)));
244 op::Add(op::Reshape(param0), param1));
[all …]
Ddot_decomposer_test.cc47 op::Reshape(AllOf(op::Dot(op::Reshape(), op::Reshape(), in TEST_F()
92 op::Reshape(AllOf(op::Dot(op::Reshape(), op::Reshape(), in TEST_F()
117 op::Reshape(AllOf(op::Dot(op::Reshape(), op::Reshape(), in TEST_F()
Ddynamic_index_splitter_test.cc55 op::Reshape(op::Slice(op::Parameter(1))), in TEST_F()
56 op::Reshape(op::Slice(op::Parameter(1))), in TEST_F()
57 op::Reshape(op::Slice(op::Parameter(1))))); in TEST_F()
93 op::Reshape(op::Slice(op::Parameter(1))), in TEST_F()
94 op::Reshape(op::Slice(op::Parameter(1))), in TEST_F()
95 op::Reshape(op::Slice(op::Parameter(1))))); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner_test.cc146 op::Reshape(op::DynamicSlice(op::Constant(), op::PartitionId())), in TEST_F()
168 op::Reshape(op::DynamicSlice(op::Constant(), op::PartitionId())), in TEST_F()
190 op::Reshape(op::DynamicSlice(op::Constant(), op::PartitionId())), in TEST_F()
210 AllOf(op::Copy(op::Reshape(op::Transpose(op::AllToAll(AllOf( in TEST_F()
211 op::Reshape(op::Parameter()), op::Shape("s32[4,2,1]")))))), in TEST_F()
230 AllOf(op::Copy(op::Slice(op::Reshape(AllOf(op::Transpose(op::AllToAll( in TEST_F()
231 op::Reshape(AllOf(op::Pad(), op::Shape("f32[8,16,128]"))))))))))); in TEST_F()
285 op::Reshape(op::DynamicSlice(op::Constant(), op::PartitionId())); in TEST_F()
315 op::Reshape(op::DynamicSlice(op::Constant(), op::PartitionId())), in TEST_F()
447 op::Compare(op::Add(op::Iota(), op::Broadcast(op::Reshape())), in TEST_F()
[all …]
Dcanonicalize_all_gather_for_cse_test.cc69 AllOf(op::Reshape(op::AllGather(_)), op::Shape("s32[2,8]"))); in TEST_F()
89 AllOf(op::Reshape(op::AllGather(_)), op::Shape("s32[2,8,1,1]"))); in TEST_F()
109 AllOf(op::Reshape(op::AllGather(_)), op::Shape("s32[2,8,1,1]"))); in TEST_F()
128 EXPECT_THAT(reshape, AllOf(op::AllGather(op::Reshape(op::Reshape(_))), in TEST_F()
148 EXPECT_THAT(reshape, AllOf(op::AllGather(op::Reshape(op::Reshape(_))), in TEST_F()
169 AllOf(op::Reshape(op::AllGather(_)), op::Shape("s32[2,8,1,1]"))); in TEST_F()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/
Dunroll-batch-matmul.mlir22 …// CHECK: %[[v0:.*]] = "tf.Reshape"(%arg0, %[[cst]]) : (tensor<2x3x4x5xf32>, tensor<3xi64>) -> ten…
24 …// CHECK: %[[v2:.*]] = "tf.Reshape"(%[[v1]], %[[cst_1]]) : (tensor<1x4x5xf32>, tensor<2xi64>) -> t…
26 …// CHECK: %[[v4:.*]] = "tf.Reshape"(%[[v3]], %[[cst_1]]) : (tensor<1x4x5xf32>, tensor<2xi64>) -> t…
28 …// CHECK: %[[v6:.*]] = "tf.Reshape"(%[[v5]], %[[cst_1]]) : (tensor<1x4x5xf32>, tensor<2xi64>) -> t…
30 …// CHECK: %[[v8:.*]] = "tf.Reshape"(%[[v7]], %[[cst_1]]) : (tensor<1x4x5xf32>, tensor<2xi64>) -> t…
32 …// CHECK: %[[v10:.*]] = "tf.Reshape"(%[[v9]], %[[cst_1]]) : (tensor<1x4x5xf32>, tensor<2xi64>) -> …
34 …// CHECK: %[[v12:.*]] = "tf.Reshape"(%[[v11]], %[[cst_1]]) : (tensor<1x4x5xf32>, tensor<2xi64>) ->…
36 …// CHECK: %[[v13:.*]] = "tf.Reshape"(%arg1, %[[cst_2]]) : (tensor<2x3x5x6xf32>, tensor<3xi64>) -> …
38 …// CHECK: %[[v15:.*]] = "tf.Reshape"(%[[v14]], %[[cst_10]]) : (tensor<1x5x6xf32>, tensor<2xi64>) -…
40 …// CHECK: %[[v17:.*]] = "tf.Reshape"(%[[v16]], %[[cst_10]]) : (tensor<1x5x6xf32>, tensor<2xi64>) -…
[all …]
Dtf_optimize.mlir40 …%98 = "tf.Reshape"(%97, %cst_2) : (tensor<1x8x6x1x6x1x1x18xbf16>, tensor<4xi64>) -> tensor<8x6x6x1…
45 …// CHECK: %[[RESHAPE:.*]] = "tf.Reshape"(%arg0, %[[CST]]) : (tensor<1x8x1x1x1x1x1x18xbf16>, tensor…
55 …%98 = "tf.Reshape"(%97, %cst_2) : (tensor<7x1x8x6x1x6x1x1x18xbf16>, tensor<5xi64>) -> tensor<7x8x6…
60 …// CHECK: %[[RESHAPE:.*]] = "tf.Reshape"(%arg0, %[[CST]]) : (tensor<1x8x1x1x1x1x1x18xbf16>, tensor…
70 …%98 = "tf.Reshape"(%97, %cst_2) : (tensor<1x1x6x1x6x1x1x18xbf16>, tensor<5xi64>) -> tensor<1x6x1x6…
75 …// CHECK: %[[RESHAPE:.*]] = "tf.Reshape"(%arg0, %[[CST]]) : (tensor<1x1x1x1x1x1x1x18xbf16>, tensor…
85 …%98 = "tf.Reshape"(%97, %cst_2) : (tensor<1x8x6x1x6x1x1x18xbf16>, tensor<4xi64>) -> tensor<8x6x6x1…
91 …// CHECK: %[[RESHAPE:.*]] = "tf.Reshape"(%[[BROADCAST]], %[[CST1]]) : (tensor<1x8x6x1x6x1x1x18xbf1…
102 …%98 = "tf.Reshape"(%97, %cst_2) : (tensor<1x1x6x1x6x1x1x18xbf16>, tensor<7xi64>) -> tensor<1x3x2x1…
108 …// CHECK: %[[RESHAPE:.*]] = "tf.Reshape"(%[[BROADCAST]], %[[CST1]]) : (tensor<1x1x6x1x6x1x1x18xbf1…
[all …]
Deinsum.mlir40 …// CHECK: %[[v1:.*]] = "tf.Reshape"(%arg1, %[[cst_1]]) : (tensor<5x2xf32>, tensor<3xi64>) -> tenso…
42 …// CHECK: %[[v3:.*]] = "tf.Reshape"(%[[v2]], %[[cst_2]]) : (tensor<5x7x1xf32>, tensor<2xi64>) -> t…
75 …// CHECK: %[[v0:.*]] = "tf.Reshape"(%arg0, %[[cst]]) : (tensor<2x5x7x3xf32>, tensor<3xi64>) -> ten…
76 …// CHECK: %[[v1:.*]] = "tf.Reshape"(%arg1, %[[cst_1]]) : (tensor<7x3x4xf32>, tensor<2xi64>) -> ten…
86 …// CHECK: %[[v0:.*]] = "tf.Reshape"(%arg1, %[[cst]]) : (tensor<5x6x2xf32>, tensor<2xi64>) -> tenso…
88 …// CHECK: %[[v2:.*]] = "tf.Reshape"(%[[v1]], %[[cst_1]]) : (tensor<3x4x12xf32>, tensor<4xi64>) -> …
98 …// CHECK: %[[v0:.*]] = "tf.Reshape"(%arg0, %[[cst]]) : (tensor<2x5x7xf32>, tensor<4xi64>) -> tenso…
100 …// CHECK: %[[v2:.*]] = "tf.Reshape"(%[[v1]], %[[cst_1]]) : (tensor<2x5x1x3xf32>, tensor<3xi64>) ->…
112 …// CHECK: %[[v1:.*]] = "tf.Reshape"(%arg0, %[[cst_1]]) : (tensor<2x5x7xf32>, tensor<4xi64>) -> ten…
114 …// CHECK: %[[v3:.*]] = "tf.Reshape"(%[[v2]], %[[cst_2]]) : (tensor<2x5x1x3xf32>, tensor<3xi64>) ->…
[all …]
/external/tensorflow/tensorflow/compiler/xla/tests/
Dreshape_test.cc108 auto reshape = Reshape(/*operand=*/parameter, /*dimensions=*/{0, 1}, in XLA_TEST_P()
126 Reshape(/*operand=*/a, /*dimensions=*/{}, /*new_sizes=*/{1}); in XLA_TEST_P()
212 Reshape(/*operand=*/parameter, /*dimensions=*/{0}, in XLA_TEST_P()
228 Reshape(/*operand=*/parameter, /*dimensions=*/{0}, in XLA_TEST_P()
244 Reshape(/*operand=*/parameter, /*dimensions=*/{0, 1}, in XLA_TEST_P()
260 Reshape(/*operand=*/parameter, /*dimensions=*/{0, 1}, in XLA_TEST_P()
278 Reshape(/*operand=*/parameter, /*dimensions=*/{1, 0}, in XLA_TEST_P()
327 Reshape(/*operand=*/parameter, /*dimensions=*/{0, 1}, in XLA_TEST_P()
342 Reshape(/*operand=*/parameter, /*dimensions=*/{0, 1, 2, 3}, in XLA_TEST_P()
359 Reshape(/*operand=*/parameter, /*dimensions=*/{0, 1}, in XLA_TEST_P()
[all …]
Dconvolution_test.cc277 auto input_r5 = input_r1.Reshape(input_dims).ConsumeValueOrDie(); in XLA_TEST_F()
282 auto filter_r5 = filter_r1.Reshape(filter_dims).ConsumeValueOrDie(); in XLA_TEST_F()
287 auto expected_r5 = expected_r1.Reshape({1, 3, 1, 2, 3}).ConsumeValueOrDie(); in XLA_TEST_F()
340 auto input_r4 = input_r1.Reshape(input_dims).ConsumeValueOrDie(); in RunTest()
345 auto filter_r4 = filter_r1.Reshape(filter_dims).ConsumeValueOrDie(); in RunTest()
349 auto expected_r4 = expected_r1.Reshape({1, 1, 1, 3}).ConsumeValueOrDie(); in RunTest()
400 auto input_r4 = input_r1.Reshape(input_dims).ConsumeValueOrDie(); in RunTest()
405 auto filter_r4 = filter_r1.Reshape(filter_dims).ConsumeValueOrDie(); in RunTest()
413 auto expected_r4 = expected_r1.Reshape({1, 1, 1, 15}).ConsumeValueOrDie(); in RunTest()
466 auto input_r4 = input_r1.Reshape(input_dims).ConsumeValueOrDie(); in RunTest()
[all …]
Dreshape_motion_test.cc49 auto c = Reshape(a, {6}); in TEST_F()
50 auto d = Reshape(b, {6}); in TEST_F()
/external/tensorflow/tensorflow/lite/delegates/xnnpack/
Dreshape_test.cc29 TEST(Reshape, 4DShapeAsInput) {
50 TEST(Reshape, 4DShapeAsParam) {
71 TEST(Reshape, 3DShapeAsInput) {
92 TEST(Reshape, 3DShapeAsParam) {
113 TEST(Reshape, 2DShapeAsInput) {
133 TEST(Reshape, 2DShapeAsParam) {
153 TEST(Reshape, 1DShapeAsInput) {
171 TEST(Reshape, 1DShapeAsParam) {
189 TEST(Reshape, 0D) {
200 TEST(Reshape, MultiThreading) { in TEST() argument
/external/tensorflow/tensorflow/compiler/jit/tests/
Dopens2s_gnmt_mixed_precision.golden_summary42 Reshape 2
80 Reshape 2
99 Reshape 8
124 Reshape 3
130 Reshape 3
225 Reshape 86
258 Reshape 9
298 Reshape 9
341 Reshape 76
/external/llvm-project/mlir/examples/toy/Ch7/mlir/
DToyCombine.td31 // Reshape(Reshape(x)) = Reshape(x)
42 // Reshape(Constant(x)) = x'
56 // Reshape(x) = x, where input and output shapes are identical
/external/llvm-project/mlir/examples/toy/Ch6/mlir/
DToyCombine.td31 // Reshape(Reshape(x)) = Reshape(x)
42 // Reshape(Constant(x)) = x'
56 // Reshape(x) = x, where input and output shapes are identical
/external/llvm-project/mlir/examples/toy/Ch3/mlir/
DToyCombine.td31 // Reshape(Reshape(x)) = Reshape(x)
42 // Reshape(Constant(x)) = x'
56 // Reshape(x) = x, where input and output shapes are identical
/external/llvm-project/mlir/examples/toy/Ch5/mlir/
DToyCombine.td31 // Reshape(Reshape(x)) = Reshape(x)
42 // Reshape(Constant(x)) = x'
56 // Reshape(x) = x, where input and output shapes are identical
/external/llvm-project/mlir/examples/toy/Ch4/mlir/
DToyCombine.td31 // Reshape(Reshape(x)) = Reshape(x)
42 // Reshape(Constant(x)) = x'
56 // Reshape(x) = x, where input and output shapes are identical
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dscoped_allocator_optimizer_test.cc105 Output r1 = ops::Reshape(s.WithOpName("r1"), a1, {1, 4}); in BuildAbsGraph()
106 Output r2 = ops::Reshape(s.WithOpName("r2"), a2, {4, 1}); in BuildAbsGraph()
147 Output r1 = ops::Reshape(s.WithOpName("r1"), a1, {1, 4}); in BuildAbsGraphWithInputDependencies()
148 Output r2 = ops::Reshape(s.WithOpName("r2"), a2, {4, 1}); in BuildAbsGraphWithInputDependencies()
149 Output r3 = ops::Reshape(s.WithOpName("r3"), a3, {4, 1}); in BuildAbsGraphWithInputDependencies()
182 Output o1 = ops::Reshape(s.WithOpName("o1"), a1, {1, 4}); in BuildAbsGraphWithInputAndOutputControlEdges()
183 Output o2 = ops::Reshape(s.WithOpName("o2"), a2, {4, 1}); in BuildAbsGraphWithInputAndOutputControlEdges()
228 Output r1 = ops::Reshape(root_scope.WithOpName("r1"), a1, {1, 4}); in BuildGraphWithMultipleScopes()
229 Output r2 = ops::Reshape(root_scope.WithOpName("r2"), a2, {4, 1}); in BuildGraphWithMultipleScopes()
236 Output r3 = ops::Reshape(sub_scope.WithOpName("r3"), a3, {1, 4}); in BuildGraphWithMultipleScopes()
[all …]
/external/tensorflow/tensorflow/lite/delegates/gpu/gl/kernels/
Dreshape_test.cc33 TEST(Reshape, 1x2x3To3x2x1) {
55 TEST(Reshape, 3x1x2To2x1x3) {
77 TEST(Reshape, 1x1x4To2x2x1) {
98 TEST(Reshape, BatchIsUnsupported) { in TEST() argument
/external/tensorflow/tensorflow/core/api_def/java_api/
Dapi_def_Reshape.pbtxt2 graph_op_name: "Reshape"
4 name: "Reshape"
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Ddiag_op.cc99 input = xla::Reshape(input, {size}); in Compile()
108 diag = xla::Reshape(diag, new_dims); in Compile()
150 xla::XlaOp reshape_input = xla::Reshape(input, {new_size, new_size}); in Compile()
152 xla::Reshape(is_gpu_ ? xla::GetMatrixDiagonalViaGather(reshape_input) in Compile()
Dwhere_op.cc48 xla::XlaOp reshaped_condition = xla::Reshape(condition, {flattened_size}); in Compile()
65 xla::XlaOp reshaped = xla::Reshape(iota, {flattened_size}); in Compile()
77 to_concat.push_back(xla::Reshape(index_single_dim, {flattened_size, 1})); in Compile()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dconst_analysis_test.cc44 auto c = ops::Reshape(root, arg2, b); in TEST()
75 auto a = ops::Reshape(root, arg0, arg1); in TEST()
76 auto b = ops::Reshape(root, arg2, a); in TEST()
157 Output reshape = ops::Reshape(root, arg1, add); in TEST()
181 Output reshape = ops::Reshape(root, arg1, add); in TEST()

1234567891011