| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | identity_n_op_py_test.py | 32 array_ops.identity_n([[1, 2, 3, 4, 5, 6],
45 array_ops.identity_n([inp0, inp1, inp2]))
55 [value] = self.evaluate(array_ops.identity_n([source]))
64 self.assertEqual(shape, array_ops.identity_n([tensor])[0].get_shape())
65 self.assertEqual(shape, array_ops.identity_n([array_2x3])[0].get_shape())
|
| D | functional_ops_test.py | 1234 outputs = array_ops.identity_n(outputs)
1275 outputs = array_ops.identity_n(outputs)
|
| D | resource_variable_ops_test.py | 327 identity = array_ops.identity_n([c, v.handle])
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | model_pruner.cc | 275 string NewIdentityFromIdentityN(int pos, const NodeDef& identity_n, in NewIdentityFromIdentityN() argument
280 strings::StrCat(identity_n.name(), "-", pos, "-grappler-ModelPruner"); in NewIdentityFromIdentityN()
286 .Input(identity_n.input(pos), 0, in NewIdentityFromIdentityN()
287 identity_n.attr().at("T").list().type(pos)) in NewIdentityFromIdentityN()
288 .Device(identity_n.device()) in NewIdentityFromIdentityN()
|
| D | layout_optimizer_test.cc | 1191 auto identity_n = ops::IdentityN(s.WithOpName("identity_n"), {vector, conv}); in TEST_F() local
1192 auto add = ops::Add(s.WithOpName("add"), identity_n[0], identity_n[1]); in TEST_F()
|
| D | generic_layout_optimizer_transposer_test.cc | 338 auto identity_n = in CreateSimpleIdentityN() local
342 ops::Identity(scope.WithOpName("conv2d_1_output"), identity_n.output[0]); in CreateSimpleIdentityN()
344 ops::Identity(scope.WithOpName("conv2d_2_output"), identity_n.output[1]); in CreateSimpleIdentityN()
346 ops::Identity(scope.WithOpName("a_output"), identity_n.output[2]); in CreateSimpleIdentityN()
348 ops::Identity(scope.WithOpName("b_output"), identity_n.output[3]); in CreateSimpleIdentityN()
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_IdentityN.pbtxt | 14 y, _ = identity_n([f(x), x])
|
| /external/tensorflow/tensorflow/lite/testing/op_tests/ |
| D | identity.py | 65 identity_outputs = tf.identity_n(inputs_doubled)
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/tf_saved_model/ |
| D | multi_arguments_results_v1.py | 59 [t, s] = array_ops.identity_n([t, s])
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | nary_ops_test.py | 93 self._testNAryLists(array_ops.identity_n,
96 self._testNAryLists(array_ops.identity_n,
102 self._testNAryLists(array_ops.identity_n,
|
| D | xla_device_test.py | 65 y, = array_ops.identity_n([x])
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | breakup-islands.cc | 140 auto identity_n = builder.create<TF::IdentityNOp>( in PopulateEmptyIsland() local
142 for (auto it : llvm::enumerate(identity_n.getResults())) in PopulateEmptyIsland()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | build_xla_ops_pass.cc | 455 ops::IdentityN identity_n(root.WithOpName("int32_id_n"), int32_inputs); in PredicateInt32Inputs() local
457 identity_n.operation.node()); in PredicateInt32Inputs()
460 TF_RETURN_IF_ERROR(root.graph()->UpdateEdge(identity_n[i].node(), i, n, in PredicateInt32Inputs()
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | pywrap_tfe_test.py | 166 array_ops.identity_n([a_2_by_2, b_2_by_2]),
185 z2 = array_ops.identity_n([a_2_by_2, b_2_by_2])
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/graphdef2mlir/ |
| D | multi-output-feeds.pbtxt | 16 # z0, z1, z2 = tf.identity_n((w, x, y))
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | custom_gradient.py | 419 all_tensors = array_ops.identity_n(all_tensors)
|
| D | gradients_test.py | 449 c, d = array_ops.identity_n([a, b])
479 c, d = array_ops.identity_n([a, b])
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | tf_passes.td | 108 …%identity_n:2 = "tf.IdentityN"(%arg1, %island_0_result) : (tensor<i32>, tensor<i32>) -> (tensor<i3…
109 tf_executor.yield %identity_n#0
122 …%identity_n:2 = "tf.IdentityN"(%arg1, %identity) : (tensor<i32>, tensor<i32>) -> (tensor<i32>, ten…
123 return %identity, %identity_n#0 : tensor<i32>, tensor<i32>
|
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | array_test.py | 361 return array_ops.identity_n([x, array_ops.gather(x, i)])
|
| D | pfor.py | 1177 reduction_op = array_ops.identity_n(pl_outputs)[0].op
2064 outputs = array_ops.identity_n([x.t for x in pfor_input.inputs])
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.pbtxt | 736 name: "identity_n"
|
| /external/tensorflow/tensorflow/python/grappler/ |
| D | layout_optimizer_test.py | 1907 outputs = array_ops.identity_n([y1, y])
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.pbtxt | 1456 name: "identity_n"
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_generated_ops.td | 5348 y, _ = identity_n([f(x), x])
|