Home
last modified time | relevance | path

Searched refs:shape_invariant (Results 1 – 14 of 14) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/
Ddrop_while_shape_invariant.mlir4 // CHECK-NOT: shape_invariant
6 …arg0) {cond = @while_cond, body = @while_body, is_stateless = false, shape_invariant} : (tensor<4x…
16 }) {is_stateless = false, shape_invariant} : (tensor<4xf32>) -> (tensor<*xf32>)
Dshape_inference.mlir860 … = @while_shape_invariant_body_func_propagate, is_stateless = false, shape_invariant} : (tensor<4x…
875 // CHECK-NEXT: shape_invariant
878 …}) {is_stateless = false, shape_invariant} : (tensor<4xf32>, tensor<!tf.resource<tensor<4xf32>>>, …
908 …hile_shape_invariant_body_func_different_dims, is_stateless = false, shape_invariant} : (tensor<1x…
923 // CHECK-NEXT: shape_invariant
926 }) {is_stateless = false, shape_invariant} : (tensor<1x2x3xf32>) -> tensor<1x8x3xf32>
956 …ape_invariant_body_func_body_result_propagate, is_stateless = false, shape_invariant} : (tensor<*x…
971 // CHECK-NEXT: shape_invariant
974 …}) {is_stateless = false, shape_invariant} : (tensor<*x!tf.resource<tensor<f32>>>) -> tensor<*x!tf…
Dcluster_tf_ops_pass.mlir15 …arg0) {cond = @while_cond, body = @while_body, is_stateless = false, shape_invariant, device="/job…
Dtf-ops.mlir2005 …%0 = "tf.While"(%arg0) {cond = @cond, body = @body, is_stateless = false, shape_invariant} : (tens…
2297 }) {is_stateless = false, shape_invariant} : (tensor<1x2x3xf32>) -> tensor<1x8x3xf32>
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/mlir2graphdef/
Dfunctional-while-ops.mlir7 …le"(%arg0, %arg1) {body = @body, cond = @cond, is_stateless = false, shape_invariant} : (tensor<i3…
40 // CHECK-NOT: shape_invariant
59 // CHECK-NOT: shape_invariant
77 // CHECK-NOT: shape_invariant
/external/tensorflow/tensorflow/python/autograph/operators/
Dcontrol_flow_deprecated_py2.py138 name, check_shape, init, entry, exit_, shape_invariant): argument
170 if shape_invariant is None:
179 if not _is_subshape(init_shape, shape_invariant):
183 shape_invariant))
184 if not _is_subshape(exit_shape, shape_invariant):
188 name, exit_shape, shape_invariant))
Dcontrol_flow.py213 name, check_shape, init, entry, exit_, shape_invariant): argument
248 if shape_invariant is None:
257 if not _is_subshape(init_shape, shape_invariant):
261 shape_invariant))
262 if not _is_subshape(exit_shape, shape_invariant):
266 name, exit_shape, shape_invariant))
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dfunctional_control_flow_to_regions.cc120 while_op.shape_invariant()); in ConvertWhileOp()
Dregion_control_flow_to_functional.cc427 while_region.shape_invariant()); in ConvertWhileOp()
Dshape_inference.cc1000 if (!op.shape_invariant()) in InferShapeForWhile()
1340 if (while_op.shape_invariant()) { in PropagateShapeIntoAttachedFunctions()
1376 if (while_op.shape_invariant()) { in PropagateShapeIntoAttachedRegions()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/graphdef2mlir/
Dfunctional-while-ops.pbtxt9 # CHECK-DAG: "tf.While"{{.*}} is_stateless = false{{.*}} shape_invariant{{.*}} -> (tensor<i32>, ten…
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops_n_z.cc2838 bool shape_invariant) { in VerifyWhileTypes() argument
2871 if (!shape_invariant && in VerifyWhileTypes()
2916 op.shape_invariant()))) in Verify()
2942 op.shape_invariant()))) in Verify()
Dtf_ops.td711 UnitAttr:$shape_invariant
791 UnitAttr:$shape_invariant
/external/tensorflow/tensorflow/compiler/mlir/lite/tests/
Dlower-static-tensor-list.mlir342 …d, device = "", is_stateless = true, parallel_iterations = 10 : i64, shape_invariant} : (tensor<i3…