Home
last modified time | relevance | path

Searched refs:identity_op (Results 1 – 25 of 27) sorted by relevance

12

/external/tensorflow/tensorflow/c/experimental/ops/
Darray_ops.cc28 AbstractOperationPtr identity_op(ctx->CreateOperation()); in Identity() local
30 identity_op->Reset("Identity", /*raw_device_name=*/nullptr)); in Identity()
31 TF_RETURN_IF_ERROR(MaybeSetOpName(identity_op.get(), name)); in Identity()
32 TF_RETURN_IF_ERROR(identity_op->AddInput(inputs[0])); in Identity()
34 return identity_op->Execute(outputs, &num_retvals); in Identity()
/external/tensorflow/tensorflow/lite/toco/graph_transformations/tests/
Dfuse_binary_into_following_affine_test.cc135 auto identity_op = new TensorFlowIdentityOperator; in TEST_F() local
136 identity_op->inputs = {"MulOutput"}; in TEST_F()
137 identity_op->outputs = {"AnotherOutput"}; in TEST_F()
138 model_->operators.push_back(std::unique_ptr<Operator>(identity_op)); in TEST_F()
/external/tensorflow/tensorflow/lite/python/
Dop_hint.py286 identity_op = _array_ops.identity(arg, name=name)
289 identity_op.op._set_attr(
293 identity_op.op._set_attr(
297 identity_op.op._set_attr(
299 identity_op.op._set_attr(OpHint.FUNCTION_LEVEL_ATTR,
302 identity_op.op._set_attr(
309 identity_op.op._set_attr(
313 identity_op.op._set_attr(
317 return identity_op
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dreadonly_references_to_resources.cc71 auto identity_op = op->getUsers().begin(); in GetNodeNameFromClassAttr() local
72 classes_attr = identity_op->getAttrOfType<ArrayAttr>(kClassAttr); in GetNodeNameFromClassAttr()
/external/tensorflow/tensorflow/python/debug/lib/
Ddebug_gradients.py159 identity_op = (
164 debug_grad_identity = identity_op(input_tensor, name=grad_debug_op_name)
/external/crosvm/bin/
Dclippy66 identity_op
/external/rust/crates/ring/src/
Dlib.rs52 clippy::identity_op,
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dremapper.cc1420 NodeDef identity_op; in IsCpuCompatibleDataType() local
1421 identity_op.set_op("Identity"); in IsCpuCompatibleDataType()
1422 identity_op.set_name(activation.name()); in IsCpuCompatibleDataType()
1423 identity_op.set_device(fused_batch_norm.device()); in IsCpuCompatibleDataType()
1424 identity_op.add_input(fused_batch_norm.name()); in IsCpuCompatibleDataType()
1425 (*identity_op.mutable_attr())["T"] = attrs->at("T"); in IsCpuCompatibleDataType()
1431 mutation->AddNode(std::move(identity_op), &status); in IsCpuCompatibleDataType()
/external/tensorflow/tensorflow/python/ops/
Dwhile_v2.py833 identity_op = t.consumers()[0]
834 if (identity_op.type == "Identity" and
835 any(identity_op.outputs[0] is t for t in tensor.graph.outputs)):
836 return identity_op.outputs[0]
/external/rust/crates/futures-util/src/lock/
Dmutex.rs69 #[allow(clippy::identity_op)] // https://github.com/rust-lang/rust-clippy/issues/3445
/external/crosvm/sys_util/src/
Dsock_ctrl_msg.rs433 #[allow(clippy::erasing_op, clippy::identity_op)]
/external/tensorflow/tensorflow/core/kernels/data/
DBUILD355 "//tensorflow/core/kernels:identity_op",
443 "//tensorflow/core/kernels:identity_op",
773 "//tensorflow/core/kernels:identity_op",
/external/tensorflow/tensorflow/core/common_runtime/
DBUILD2086 "//tensorflow/core/kernels:identity_op",
2208 "//tensorflow/core/kernels:identity_op",
2242 "//tensorflow/core/kernels:identity_op",
2285 "//tensorflow/core/kernels:identity_op",
2320 "//tensorflow/core/kernels:identity_op",
/external/rust/crates/half/src/
Dbfloat.rs990 #[allow(clippy::erasing_op, clippy::identity_op)]
1085 #[allow(clippy::erasing_op, clippy::identity_op)]
Dbinary16.rs1285 #[allow(clippy::erasing_op, clippy::identity_op)]
1380 #[allow(clippy::erasing_op, clippy::identity_op)]
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_compiler.cc191 auto identity_op = [builder](xla::XlaOp op) { in BuildComputation() local
242 value = identity_op(value); in BuildComputation()
344 handle = identity_op(handle); in BuildComputation()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
DBUILD54 "identity_op.cc",
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
DBUILD402 "//tensorflow/core/kernels:identity_op",
/external/tensorflow/tensorflow/core/distributed_runtime/
DBUILD689 "//tensorflow/core/kernels:identity_op",
/external/tensorflow/tensorflow/core/kernels/
DBUILD971 ":identity_op",
1087 name = "identity_op",
1088 prefix = "identity_op",
1543 ":identity_op",
2002 ":identity_op",
5816 "identity_op.cc",
5817 "identity_op.h",
/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
Dlegalize_tf.cc1954 auto identity_op = in matchAndRewrite() local
1956 val = identity_op.getResult(); in matchAndRewrite()
Dlegalize_common.cc2325 auto identity_op = in convertReduceOpCommon() local
2327 val = identity_op.getResult(); in convertReduceOpCommon()
/external/tensorflow/tensorflow/core/grappler/costs/
Dop_level_cost_estimator_test.cc2042 for (auto identity_op : identity_ops) { in TEST_F() local
2043 OpContext op_context = DescribeUnaryOp(identity_op, kTensorSize); in TEST_F()
/external/rust/crates/nix/src/sys/socket/
Daddr.rs410 #[allow(clippy::identity_op)] // More readable this way
/external/tensorflow/tensorflow/compiler/jit/
DBUILD264 "//tensorflow/core/kernels:identity_op",

12