Home
last modified time | relevance | path

Searched defs:assign_op (Results 1 – 10 of 10) sorted by relevance

/external/eigen/Eigen/src/Core/functors/
DAssignmentFunctors.h21 template<typename DstScalar,typename SrcScalar> struct assign_op { struct
23 EIGEN_EMPTY_STRUCT_CTOR(assign_op) in EIGEN_EMPTY_STRUCT_CTOR() argument
32 template<typename DstScalar> struct assign_op<DstScalar,void> {}; argument
/external/tensorflow/tensorflow/python/kernel_tests/
Ddense_update_ops_no_tsan_test.py79 def run_assign(assign_op): argument
146 def run_assign(assign_op): argument
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dlegalize_variables.cc71 TF::AssignVariableOp assign_op, in GetAssignVariableOp()
108 auto assign_op = llvm::cast<TF::AssignVariableOp>(op); in matchAndRewrite() local
/external/tensorflow/tensorflow/c/experimental/saved_model/core/ops/
Dvariable_ops.cc74 ImmediateOpPtr assign_op(ctx->CreateOperation()); in AssignVariable() local
/external/tensorflow/tensorflow/compiler/jit/
Dbuild_xla_ops_pass_test.cc112 ops::AssignVariableOp assign_op(scope.WithOpName("Assignee_" + id), in MakeWrite() local
Dresource_operation_safety_analysis_test.cc54 ops::AssignVariableOp assign_op(scope.WithOpName("Assignee" + id), var_handle, in MakeWrite() local
Dmark_for_compilation_pass_test.cc669 ops::AssignVariableOp assign_op(scope.WithOpName("Assignment" + id), in MakeWrite() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtpu_merge_variables_with_execute.cc250 auto assign_op = llvm::dyn_cast<TF::AssignVariableOp>(*result.user_begin()); in BuildVariableAccessInfo() local
/external/tensorflow/tensorflow/core/common_runtime/
Dquantize_training.cc301 Node* assign_op; in AddRestoreVariableSubgraphs() local
/external/tensorflow/tensorflow/compiler/mlir/lite/
Dflatbuffer_export.cc1104 } else if (auto assign_op = in BuildOperator() local