| /external/eigen/Eigen/src/Core/functors/ |
| D | AssignmentFunctors.h | 21 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/variables/ |
| D | dense_update_ops_no_tsan_test.py | 75 def run_assign(assign_op): argument
142 def run_assign(assign_op): argument
|
| /external/tensorflow/tensorflow/dtensor/mlir/ |
| D | restore_shape_inference.cc | 117 module.walk([&](mlir::TF::AssignVariableOp assign_op) { in PropagateShapeInformationFromAssignVariableOp()
|
| /external/tensorflow/tensorflow/c/experimental/saved_model/core/ops/ |
| D | variable_ops.cc | 73 ImmediateOpPtr assign_op(ctx->CreateOperation()); in AssignVariable() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | hoist_replicate_invariant_resource_writes.cc | 91 auto assign_op = indexed_assign.value(); in MoveTailWritesAfterReplicate() local
|
| D | mark_input_output_aliases.cc | 58 auto assign_op = llvm::dyn_cast_or_null<TF::AssignVariableOp>( in BuildAliasingInfo() local
|
| D | tpu_merge_variables_with_execute.cc | 271 auto assign_op = llvm::dyn_cast<TF::AssignVariableOp>(*result.user_begin()); in BuildVariableAccessInfo() local
303 if (auto assign_op = llvm::dyn_cast<TF::AssignVariableOp>(&op)) { in BuildVariableAccessInfo() local
|
| D | shape_inference.cc | 1021 TF::AssignVariableOp assign_op = llvm::dyn_cast<TF::AssignVariableOp>(use); in InferShapeForRestore() local
1285 if (AssignVariableOp assign_op = dyn_cast<AssignVariableOp>(def)) { in InferShapeForVarHandleOp() local
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | build_xla_ops_pass_test.cc | 111 ops::AssignVariableOp assign_op(scope.WithOpName("Assignee_" + id), in MakeWrite() local
|
| D | resource_operation_safety_analysis_test.cc | 54 ops::AssignVariableOp assign_op(scope.WithOpName("Assignee" + id), var_handle, in MakeWrite() local
|
| D | mark_for_compilation_pass_test.cc | 708 ops::AssignVariableOp assign_op(scope.WithOpName("Assignment" + id), in MakeWrite() local
|
| /external/tensorflow/tensorflow/dtensor/mlir/expansions/ |
| D | save_restore_spmd_expander.cc | 821 if (auto assign_op = llvm::dyn_cast_or_null<mlir::TF::AssignVariableOp>( in GetLayoutsFromAssignVariableOps() local
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | quantize_training.cc | 301 Node* assign_op; in AddRestoreVariableSubgraphs() local
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | arithmetic_optimizer_test.cc | 4735 ops::AssignVariableOp assign_op(s.WithOpName("assign_var_handle"), var, in TEST_F() local
|