Home
last modified time | relevance | path

Searched refs:ShapeN (Results 1 – 25 of 34) sorted by relevance

12

/external/tensorflow/tensorflow/core/api_def/java_api/
Dapi_def_ShapeN.pbtxt2 graph_op_name: "ShapeN"
/external/tensorflow/tensorflow/core/api_def/python_api/
Dapi_def_ShapeN.pbtxt2 graph_op_name: "ShapeN"
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_ShapeN.pbtxt2 graph_op_name: "ShapeN"
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
DShapeN.pbtxt2 name: "ShapeN"
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/
DShapeN.pbtxt2 name: "ShapeN"
/external/tensorflow/tensorflow/compiler/jit/tests/
Dkeras_imagenet_main_graph_mode.golden_summary43 ShapeN 50
Dkeras_imagenet_main_graph_mode.pbtxt28356 name: "ConstantFolding/training/SGD/gradients/conv1_1/Conv2D_grad/ShapeN-matshapes-1"
30005 name: "training/SGD/gradients/res2a_branch1_1/Conv2D_grad/ShapeN"
30006 op: "ShapeN"
30030 name: "training/SGD/gradients/res2a_branch2a_1/Conv2D_grad/ShapeN"
30031 op: "ShapeN"
30175 name: "ConstantFolding/training/SGD/gradients/res2a_branch1_1/Conv2D_grad/ShapeN-matshapes-1"
30177 input: "^training/SGD/gradients/res2a_branch1_1/Conv2D_grad/ShapeN"
30201 name: "ConstantFolding/training/SGD/gradients/res2a_branch2a_1/Conv2D_grad/ShapeN-matshapes-1"
30203 input: "^training/SGD/gradients/res2a_branch2a_1/Conv2D_grad/ShapeN"
30475 name: "training/SGD/gradients/res2a_branch2b_1/Conv2D_grad/ShapeN"
[all …]
Dkeras_imagenet_main.pbtxt28916 …name: "ConstantFolding/training/LossScaleOptimizer/gradients/conv1_1/Conv2D_grad/ShapeN-matshapes-…
35145 …name: "ConstantFolding/training/LossScaleOptimizer/gradients/res2a_branch2a_1/Conv2D_grad/ShapeN-m…
35172 …name: "ConstantFolding/training/LossScaleOptimizer/gradients/res2a_branch2a_1/Conv2D_grad/ShapeN-m…
35199 …name: "ConstantFolding/training/LossScaleOptimizer/gradients/res2a_branch1_1/Conv2D_grad/ShapeN-ma…
35226 …name: "ConstantFolding/training/LossScaleOptimizer/gradients/res2a_branch1_1/Conv2D_grad/ShapeN-ma…
35483 …name: "ConstantFolding/training/LossScaleOptimizer/gradients/res2a_branch2b_1/Conv2D_grad/ShapeN-m…
35510 …name: "ConstantFolding/training/LossScaleOptimizer/gradients/res2a_branch2b_1/Conv2D_grad/ShapeN-m…
35753 …name: "ConstantFolding/training/LossScaleOptimizer/gradients/res2a_branch2c_1/Conv2D_grad/ShapeN-m…
35780 …name: "ConstantFolding/training/LossScaleOptimizer/gradients/res2a_branch2c_1/Conv2D_grad/ShapeN-m…
36036 …name: "ConstantFolding/training/LossScaleOptimizer/gradients/res2b_branch2a_1/Conv2D_grad/ShapeN-m…
[all …]
Dopens2s_gnmt_mixed_precision.golden_summary45 ShapeN 10
Dopens2s_gnmt_mixed_precision.pbtxt.gz
/external/tensorflow/tensorflow/core/grappler/verifiers/
Dstructure_verifier_test.cc67 ops::ShapeN b(s.WithOpName("b"), {a, a, a}); in TEST_F()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/
Dconstant-fold.mlir96 …%0:2 = "tf.ShapeN"(%arg0, %arg1) : (tensor<f32>, tensor<1x32x32x16xf32>) -> (tensor<0xi64>, tensor…
106 …// CHECK: %[[SHAPE13:.*]]:2 = "tf.ShapeN"(%arg1, %arg3) : (tensor<2x?x3xf32>, tensor<*xf32>) -> (t…
107 …%0:4 = "tf.ShapeN"(%arg0, %arg1, %arg2, %arg3) : (tensor<f32>, tensor<2x?x3xf32>, tensor<1x32x32x1…
118 …%0:3 = "tf.ShapeN"(%arg0, %arg1, %arg2) : (tensor<f32>, tensor<1x32x32x16xf32>, tensor<*xf32>) -> …
127 %0:1 = "tf.ShapeN"(%arg0) : (tensor<*xf32>) -> tensor<?xi64>
Dtf-ops.mlir2350 // CHECK-NEXT: "tf.ShapeN"
2351 …%0:2 = "tf.ShapeN"(%arg0, %arg1) : (tensor<1x32x32x16xf32>, tensor<*xf32>) -> (tensor<4xi32>, tens…
2359 …%0:2 = "tf.ShapeN"(%arg0, %arg0) : (tensor<1x32x32x16xf32>, tensor<1x32x32x16xf32>) -> (tensor<4xi…
2368 …%0:2 = "tf.ShapeN"(%arg0, %arg0) : (tensor<1x32x32x16xf32>, tensor<1x32x32x16xf32>) -> (tensor<4xi…
2377 …%0:2 = "tf.ShapeN"(%arg0, %arg0) : (tensor<1x32x32x16xf32>, tensor<1x32x32x16xf32>) -> (tensor<4xi…
2386 …%0:2 = "tf.ShapeN"(%arg0, %arg0) : (tensor<*xf32>, tensor<*xf32>) -> (tensor<?xi32>, tensor<2xi32>)
2395 …%0:2 = "tf.ShapeN"(%arg0, %arg0, %arg0) : (tensor<*xf32>, tensor<*xf32>, tensor<*xf32>) -> (tensor…
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dpin_to_host_optimizer_test.cc241 ops::ShapeN b(s.WithOpName("b"), {a, a, a}); in TEST_F()
Dlayout_optimizer_test.cc1034 auto shapen = ops::ShapeN(s.WithOpName("shapen"), {conv, conv}); in TEST_F()
1064 auto shapen = ops::ShapeN(s.WithOpName("shapen"), {vector, conv}); in TEST_F()
1091 auto shapen = ops::ShapeN(s.WithOpName("shapen"), {conv, i2}); in TEST_F()
Dauto_mixed_precision_test.cc339 auto clr3 = ops::ShapeN(s.WithOpName("clr3"), {clr1, clr2}); in TEST_F()
Dconstant_folding_test.cc1637 auto s = ops::ShapeN(scope.WithOpName("s"), {v1, v2, v3}); in TEST_F()
1711 auto s = ops::ShapeN(scope.WithOpName("s"), {v1, v2}); in TEST_F()
/external/tensorflow/tensorflow/go/genop/internal/
Dgenop_test.go413 func ShapeN(scope *Scope, input []tf.Output, optional ...ShapeNAttr) (output []tf.Output) {
/external/tensorflow/tensorflow/core/grappler/
Dgraph_view_test.cc38 ops::ShapeN b(s.WithOpName("b"), {a, a, a}); in TEST_F()
/external/tensorflow/tensorflow/lite/g3doc/guide/
Dop_select_allowlist.md476 * `raw_ops.ShapeN`
/external/tensorflow/tensorflow/core/common_runtime/
Dconstant_folding_test.cc452 auto shape_n = ops::ShapeN(s.WithOpName("shape_n"), {recv0, recv1}); in TEST_F()
/external/tensorflow/tensorflow/compiler/tf2xla/g3doc/
Dgpu_supported_ops.md203 `ShapeN` | `out_type={int32,int64}`<br>`T={bool,complex64,double,float…
Dcpu_supported_ops.md206 `ShapeN` | `out_type={int32,int64}`<br>`T={bool,complex64,double,float…
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties_test.cc1883 auto shp = ops::ShapeN(s.WithOpName("shapes"), {a, b}); in TEST_F()
/external/tensorflow/tensorflow/compiler/tests/
Drandomized_tests.cc2815 TEST_F(OpTest, ShapeN) { in TEST_F() argument

12