Home
last modified time | relevance | path

Searched defs:identity_op (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dreadonly_references_to_resources.cc71 auto identity_op = op->getUsers().begin(); in GetNodeNameFromClassAttr() local
/external/tensorflow/tensorflow/c/experimental/ops/
Darray_ops.cc28 AbstractOperationPtr identity_op(ctx->CreateOperation()); in Identity() local
/external/tensorflow/tensorflow/lite/toco/graph_transformations/tests/
Dfuse_binary_into_following_affine_test.cc135 auto identity_op = new TensorFlowIdentityOperator; in TEST_F() local
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_compiler.cc191 auto identity_op = [builder](xla::XlaOp op) { in BuildComputation() local
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dremapper.cc1420 NodeDef identity_op; in IsCpuCompatibleDataType() local
/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
Dlegalize_tf.cc1954 auto identity_op = in matchAndRewrite() local
Dlegalize_common.cc2325 auto identity_op = in convertReduceOpCommon() local
Dlegalize_tfl.cc2768 auto identity_op = in matchAndRewrite() local
/external/tensorflow/tensorflow/core/grappler/costs/
Dop_level_cost_estimator_test.cc2042 for (auto identity_op : identity_ops) { in TEST_F() local