Searched refs:identity_op (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/tests/ |
| D | fuse_binary_into_following_affine_test.cc | 134 auto identity_op = new TensorFlowIdentityOperator; in TEST_F() local
135 identity_op->inputs = {"MulOutput"}; in TEST_F()
136 identity_op->outputs = {"AnotherOutput"}; in TEST_F()
137 model_->operators.push_back(std::unique_ptr<Operator>(identity_op)); in TEST_F()
|
| /external/tensorflow/tensorflow/lite/python/ |
| D | op_hint.py | 274 identity_op = _array_ops.identity(arg, name=name)
277 identity_op.op._set_attr(
281 identity_op.op._set_attr(
285 identity_op.op._set_attr(
287 identity_op.op._set_attr(OpHint.FUNCTION_LEVEL_ATTR,
290 identity_op.op._set_attr(
297 identity_op.op._set_attr(
301 identity_op.op._set_attr(
305 return identity_op
|
| /external/tensorflow/tensorflow/python/debug/lib/ |
| D | debug_gradients.py | 159 identity_op = (
164 debug_grad_identity = identity_op(input_tensor, name=grad_debug_op_name)
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | while_v2.py | 629 identity_op = t.consumers()[0]
630 if (identity_op.type == "Identity" and
631 identity_op.outputs[0] in tensor.graph.outputs):
632 return identity_op.outputs[0]
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_compiler.cc | 181 auto identity_op = [builder](xla::XlaOp op) { in BuildComputation() local
223 value = identity_op(value); in BuildComputation()
291 handle = identity_op(handle); in BuildComputation()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | BUILD | 47 "identity_op.cc",
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | BUILD | 365 "//tensorflow/core/kernels:identity_op",
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | BUILD | 886 ":identity_op",
983 name = "identity_op",
984 prefix = "identity_op",
1326 ":identity_op",
1674 ":identity_op",
5510 "identity_op.cc",
5511 "identity_op.h",
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | BUILD | 675 "//tensorflow/core/kernels:identity_op",
|
| /external/tensorflow/tensorflow/contrib/makefile/ |
| D | tf_op_files.txt | 140 tensorflow/core/kernels/identity_op.cc
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | BUILD | 203 "//tensorflow/core/kernels:identity_op",
|
| /external/tensorflow/tensorflow/core/ |
| D | BUILD | 4403 "//tensorflow/core/kernels:identity_op",
4518 "//tensorflow/core/kernels:identity_op",
4550 "//tensorflow/core/kernels:identity_op",
4591 "//tensorflow/core/kernels:identity_op",
4626 "//tensorflow/core/kernels:identity_op",
|