Home
last modified time | relevance | path

Searched refs:RESHAPE (Results 1 – 25 of 37) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/
Dtf_optimize.mlir45 …// CHECK: %[[RESHAPE:.*]] = "tf.Reshape"(%arg0, %[[CST]]) : (tensor<1x8x1x1x1x1x1x18xbf16>, tensor…
46 …// CHECK: %[[BROADCAST:.*]] = "tf.BroadcastTo"(%[[RESHAPE]], %[[CST1]]) : (tensor<8x1x1x18xbf16>, …
60 …// CHECK: %[[RESHAPE:.*]] = "tf.Reshape"(%arg0, %[[CST]]) : (tensor<1x8x1x1x1x1x1x18xbf16>, tensor…
61 …// CHECK: %[[BROADCAST:.*]] = "tf.BroadcastTo"(%[[RESHAPE]], %[[CST1]]) : (tensor<1x8x1x1x18xbf16>…
75 …// CHECK: %[[RESHAPE:.*]] = "tf.Reshape"(%arg0, %[[CST]]) : (tensor<1x1x1x1x1x1x1x18xbf16>, tensor…
76 …// CHECK: %[[BROADCAST:.*]] = "tf.BroadcastTo"(%[[RESHAPE]], %[[CST1]]) : (tensor<1x1x1x1x18xbf16>…
91 …// CHECK: %[[RESHAPE:.*]] = "tf.Reshape"(%[[BROADCAST]], %[[CST1]]) : (tensor<1x8x6x1x6x1x1x18xbf1…
92 …// CHECK: return %[[BROADCAST]], %[[RESHAPE]] : tensor<1x8x6x1x6x1x1x18xbf16>, tensor<8x6x6x18xbf1…
108 …// CHECK: %[[RESHAPE:.*]] = "tf.Reshape"(%[[BROADCAST]], %[[CST1]]) : (tensor<1x1x6x1x6x1x1x18xbf1…
109 // CHECK: return %[[RESHAPE]] : tensor<1x3x2x1x3x2x18xbf16>
Dlayout_optimization_move_transposes_begin.mlir88 …// CHECK: %[[RESHAPE:.*]] = "tf.Reshape"(%[[ADD]], %[[CST_2]]) {device = ""} : (tensor<64x8x64xf32…
89 // CHECK: return %[[RESHAPE]] : tensor<512x64xf32>
/external/tensorflow/tensorflow/lite/delegates/gpu/gl/kernels/
Dreshape_test.cc47 SingleOpModel model({ToString(OperationType::RESHAPE), attr}, {input},
69 SingleOpModel model({ToString(OperationType::RESHAPE), attr}, {input},
91 SingleOpModel model({ToString(OperationType::RESHAPE), attr}, {input},
112 SingleOpModel model({ToString(OperationType::RESHAPE), attr}, {input}, in TEST()
Dregistry.cc94 insert_op(Type::RESHAPE, NewReshapeNodeShader); in Registry()
/external/tensorflow/tensorflow/compiler/mlir/hlo/tests/
Doptimize-hlo.mlir7 // CHECK: [[RESHAPE:%.+]] = "mhlo.reshape"([[SLICE]])
18 // CHECK: return [[RESHAPE]]
25 // CHECK: [[RESHAPE:%.+]] = "mhlo.reshape"(%arg1)
26 …// CHECK: [[SLICE:%.+]] = "mhlo.dynamic-slice"(%arg0, [[RESHAPE]], [[CST]], [[CST]]) {slice_sizes …
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/compile_mlir_util/
Dconstant-folding.mlir21 // CHECK: [[RESHAPE:%.*]] = f32[10,19]{1,0} reshape(f32[19,10]{1,0} [[ARG1]])
22 // CHECK: ROOT %tuple.{{[0-9]+}} = (f32[10,19]{1,0}) tuple(f32[10,19]{1,0} [[RESHAPE]])
/external/llvm-project/flang/runtime/
Dtransformational.h17 OwningPtr<Descriptor> RESHAPE(const Descriptor &source, const Descriptor &shape,
Dtransformational.cpp34 OwningPtr<Descriptor> RESHAPE(const Descriptor &source, const Descriptor &shape, in RESHAPE() function
/external/tensorflow/tensorflow/core/kernels/hexagon/
Dhexagon_ops_definitions.cc91 RESHAPE, enumerator
227 RESHAPE,
335 EmplaceOpType("Reshape", {}, SupportedOpType::RESHAPE, &op_map); in BuildOpNameToSocOpTypeMap()
/external/tensorflow/tensorflow/compiler/mlir/tosa/g3doc/
Dlegalization.md108 %out = tosa.RESHAPE(%a, new_a_shape)
126 … %out = tosa.RESHAPE(%a) {new_shape=new_a_shape} (tensor<%a.type>) -> tensor<new_a_shape, %a.dtype>
315 …%a1_reshape = tosa.RESHAPE(%input) {new_shape=a1_shape} : (tensor<%input.type>) -> tensor<a1_shape…
317 …%a3_reshape = tosa.RESHAPE(%a2_transpose) {new_shape=a3_shape} : (tensor<%a2_transpose.type>) -> t…
336 …%values:i = tosa.RESHAPE(%values:i) {new_shape=1} : (tensor<{}, %values:i.dtype>) -> tensor<{1}, %…
372 …%a2_reshape = tosa.RESHAPE(%input) {new_shape=a2_shape} : (tensor<%input.type>) -> tensor<a2_shape…
374 …%output = tosa.RESHAPE(%a3_transpose) {new_shape=a4_shape} : (tensor<%a3_transpose.type>) -> tenso…
396 …%one_reshape = tosa.RESHAPE(%one_const) {new_shape=bcast_shape} : (tensor<%one_const.type>) -> ten…
397 …%zero_reshape = tosa.RESHAPE(%zero_const) {new_shape=bcast_shape} : (tensor<%zero_const.type>) -> …
429 …%output = tosa.RESHAPE(%input) {new_shape=reshape_dims} (tensor<%input.type>) -> tensor<%output.ty…
[all …]
/external/llvm-project/mlir/include/mlir/Dialect/Tosa/Transforms/
DPasses.td19 number of dimensions. Insert RESHAPE operations to prepend dimensions
/external/llvm-project/flang/unittests/Evaluate/
Dreshape.cpp55 auto result{RESHAPE(*source, *shape, &pad)}; in main()
/external/tensorflow/tensorflow/compiler/mlir/lite/tests/mlir2flatbuffer/
Dreshape.mlir10 // CHECK-NEXT: builtin_code: RESHAPE
Dquantization.mlir17 // CHECK-NEXT: builtin_code: RESHAPE
/external/tensorflow/tensorflow/lite/delegates/gpu/common/transformations/
Dremove_noop.cc116 if (node->operation.type != ToString(OperationType::RESHAPE)) { in ApplyToNode()
Dremove_noop_test.cc220 simple_node->operation.type = ToString(OperationType::RESHAPE); in TEST()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Doperations.h79 RESHAPE, enumerator
Doperations.cc161 case OperationType::RESHAPE: in ToString()
244 {"reshape", OperationType::RESHAPE}, in OperationTypeFromString()
/external/tensorflow/tensorflow/compiler/mlir/lite/tests/
Dcanonicalize.mlir26 // CHECK: %[[RESHAPE:.*]] = "tfl.reshape"(%arg0, %[[CST]]) : (tensor<4x4x4xf32>, tensor<1xi32>) ->…
27 // CHECK: return %[[RESHAPE]]
/external/tensorflow/tensorflow/lite/schema/
Dschema_v1.fbs86 RESHAPE = 25,
Dschema_v2.fbs84 RESHAPE = 22,
Dschema_v3.fbs94 RESHAPE = 22,
/external/tensorflow/tensorflow/lite/delegates/gpu/
DREADME.md45 * `RESHAPE v1`
/external/tensorflow/tensorflow/lite/tools/benchmark/
DREADME.md285 …RESHAPE 79.440 0.002 0.002 0.000% 1.992% 0.000 0 [MobilenetV1/Predicti…
307 RESHAPE 1 0.000 0.000% 100.000% 0.000 0
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/
Dlegalize-tf.mlir2565 // CHECK-DAG: [[RESHAPE:%.+]] = "mhlo.dynamic_reshape"(%arg0, [[TOEXTENTS]])
2568 // CHECK: return [[RESHAPE]]
2586 // CHECK-DAG: [[RESHAPE:%.+]] = "mhlo.dynamic_reshape"(%arg0, [[TOEXTENTS]])
2589 // CHECK: return [[RESHAPE]]
2967 // CHECK: [[RESHAPE:%.*]] = "mhlo.reshape"([[SLICE]]) : (tensor<2x16x2xf32>) -> tensor<2x16x2xf32>
2969 // CHECK: return [[RESHAPE]] : tensor<2x16x2xf32>
3471 // CHECK-DAG: [[RESHAPE:%.+]] = "mhlo.reshape"([[CONVERT3]])
3472 // CHECK-DAG: [[IOTA:%.+]] = "mhlo.dynamic_iota"([[RESHAPE]]) {iota_dimension = 0 : i64}
3493 // CHECK-DAG: [[RESHAPE:%.+]] = "mhlo.reshape"([[CONVERT3]])
3494 // CHECK-DAG: [[IOTA:%.+]] = "mhlo.dynamic_iota"([[RESHAPE]]) {iota_dimension = 0 : i64}
[all …]

12