| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | graph_compiler_util.cc | 84 FunctionLibraryDefinition::FunctionLibraryDefinition::kArgOp) in AddArgNodes()
173 if (n->type_string() == FunctionLibraryDefinition::kArgOp) { in RewriteAndPruneGraph()
177 return errors::Aborted(FunctionLibraryDefinition::kArgOp, in RewriteAndPruneGraph()
204 if (n->type_string() == FunctionLibraryDefinition::kArgOp) { in CollectArgNodes()
211 "Multiple ", FunctionLibraryDefinition::kArgOp, in CollectArgNodes()
222 "Expected ", FunctionLibraryDefinition::kArgOp, " node with index ", in CollectArgNodes()
|
| D | resource_util.cc | 40 const char kArgOp[] = "_Arg"; variable
104 TF_RET_CHECK(n.type_string() == kArgOp); in PropagateFromArgOp()
149 if (src_node_info.op_ == kArgOp) { in UpdateResourceUsageFromFunctionBodyAnalysis()
|
| D | functionalize_while.cc | 112 const char* const kArgOp = "_Arg"; in BuildArgNode() local
114 NodeDefBuilder builder(absl::StrCat(kArgOp, index), kArgOp); in BuildArgNode()
|
| D | functionalize_cond.cc | 447 FunctionLibraryDefinition::kArgOp) in BuildArgumentNodes()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_device_ops.h | 156 Name(kArgOp).Device(DEVICE).TypeConstraint("T", TYPES), ArgOp); \
157 REGISTER_KERNEL_BUILDER(Name(kArgOp) \
163 Name(kArgOp).Device(DEVICE).TypeConstraint<Variant>("T"), ArgOp); \
222 REGISTER_KERNEL_BUILDER(Name(FunctionLibraryDefinition::kArgOp) \
|
| D | encapsulate_subgraphs_pass.cc | 142 static const char* const kArgOp = "_Arg"; variable
481 absl::StrCat(src_node->name(), "_", src_slot, "_arg"), kArgOp, in RecordArg()
1132 if (n->type_string() == kArgOp) { in GetArgTypes()
1150 if (n->type_string() == kArgOp) { in RenumberArguments()
|
| D | xla_compilation_cache.cc | 236 NodeBuilder(arg_name, FunctionLibraryDefinition::kArgOp) in CreateGraph()
|
| D | extract_outside_compilation_pass.cc | 1373 NodeDefBuilder arg_builder("key_arg", FunctionLibraryDefinition::kArgOp); in ReplaceKeyPlaceholderWithArgNode()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | function_ops.h | 24 static const char* const kArgOp = FunctionLibraryDefinition::kArgOp; variable
|
| D | function_ops.cc | 87 REGISTER_SYSTEM_KERNEL_BUILDER(Name(kArgOp).Device(DEVICE_CPU), ArgOp);
100 Name(kArgOp).Device(DEVICE_GPU).TypeConstraint<type>("T"), ArgOp);
103 TF_CALL_bool(REGISTER) REGISTER_KERNEL_BUILDER(Name(kArgOp)
112 REGISTER_KERNEL_BUILDER(Name(kArgOp)
118 REGISTER_KERNEL_BUILDER(Name(kArgOp)
125 Name(kArgOp).Device(DEVICE_GPU).TypeConstraint<Variant>("T"), ArgOp);
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | function_body.cc | 37 } else if (n->type_string() == FunctionLibraryDefinition::kArgOp || in FunctionBody()
|
| D | shape_refiner.cc | 60 constexpr char kArgOp[] = "_Arg"; variable
73 if (StringPiece(node->type_string()) == kArgOp) { in InferShapesForFunctionSubNode()
|
| D | inline_function_utils.cc | 54 static constexpr const char* const kArgOp = FunctionLibraryDefinition::kArgOp; variable
|
| D | function.cc | 57 static constexpr const char* const kArgOp = FunctionLibraryDefinition::kArgOp; variable
106 ndef.set_op(kArgOp); in AddArg()
894 (n->op_def().is_stateful() && n->type_string() != kArgOp) || in PruneFunctionBody()
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_compile_op_common.cc | 54 static constexpr char kArgOp[] = "_Arg"; variable
438 if (node->type_string() == kArgOp) { in AssignDevicesToArgsAndRetvals()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | function.cc | 45 /* static */ constexpr const char* const FunctionLibraryDefinition::kArgOp; member in tensorflow::FunctionLibraryDefinition
196 gnode->set_op(FunctionLibraryDefinition::kArgOp); in BuildInputArgIndex()
614 if (n->op() == FunctionLibraryDefinition::kArgOp || in Print()
|
| D | function.h | 370 static constexpr const char* const kArgOp = "_Arg";
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | export_graphdef.cc | 221 node_def->set_op(FunctionLibraryDefinition::kArgOp); in GetArgumentNode()
|
| D | import_model.cc | 735 op_name != FunctionLibraryDefinition::kArgOp) { in AddNodesToShapeRefiner()
826 if (StringPiece(node->type_string()) == FunctionLibraryDefinition::kArgOp) { in AddNodesToShapeRefiner()
1362 op_name != FunctionLibraryDefinition::kArgOp) { in ConvertFeedsToPlaceholders()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | captured_function.cc | 202 if (ret_input_node->def().op() == FunctionLibraryDefinition::kArgOp) { in CreateShortCircuitInfo()
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | encapsulate_tpu_computations_pass.cc | 665 FunctionLibraryDefinition::kArgOp); in MoveHeadOutsideCompilationToHost()
|