Home
last modified time | relevance | path

Searched refs:kArgOp (Results 1 – 21 of 21) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/
Dgraph_compiler_util.cc84 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()
Dresource_util.cc40 const char kArgOp[] = "_Arg"; variable
104 TF_RET_CHECK(n.type_string() == kArgOp); in PropagateFromArgOp()
149 if (src_node_info.op_ == kArgOp) { in UpdateResourceUsageFromFunctionBodyAnalysis()
Dfunctionalize_while.cc112 const char* const kArgOp = "_Arg"; in BuildArgNode() local
114 NodeDefBuilder builder(absl::StrCat(kArgOp, index), kArgOp); in BuildArgNode()
Dfunctionalize_cond.cc447 FunctionLibraryDefinition::kArgOp) in BuildArgumentNodes()
/external/tensorflow/tensorflow/compiler/jit/
Dxla_device_ops.h156 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) \
Dencapsulate_subgraphs_pass.cc142 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()
Dxla_compilation_cache.cc236 NodeBuilder(arg_name, FunctionLibraryDefinition::kArgOp) in CreateGraph()
Dextract_outside_compilation_pass.cc1373 NodeDefBuilder arg_builder("key_arg", FunctionLibraryDefinition::kArgOp); in ReplaceKeyPlaceholderWithArgNode()
/external/tensorflow/tensorflow/core/kernels/
Dfunction_ops.h24 static const char* const kArgOp = FunctionLibraryDefinition::kArgOp; variable
Dfunction_ops.cc87 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/
Dfunction_body.cc37 } else if (n->type_string() == FunctionLibraryDefinition::kArgOp || in FunctionBody()
Dshape_refiner.cc60 constexpr char kArgOp[] = "_Arg"; variable
73 if (StringPiece(node->type_string()) == kArgOp) { in InferShapesForFunctionSubNode()
Dinline_function_utils.cc54 static constexpr const char* const kArgOp = FunctionLibraryDefinition::kArgOp; variable
Dfunction.cc57 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/
Dtpu_compile_op_common.cc54 static constexpr char kArgOp[] = "_Arg"; variable
438 if (node->type_string() == kArgOp) { in AssignDevicesToArgsAndRetvals()
/external/tensorflow/tensorflow/core/framework/
Dfunction.cc45 /* 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()
Dfunction.h370 static constexpr const char* const kArgOp = "_Arg";
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dexport_graphdef.cc221 node_def->set_op(FunctionLibraryDefinition::kArgOp); in GetArgumentNode()
Dimport_model.cc735 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/
Dcaptured_function.cc202 if (ret_input_node->def().op() == FunctionLibraryDefinition::kArgOp) { in CreateShortCircuitInfo()
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Dencapsulate_tpu_computations_pass.cc665 FunctionLibraryDefinition::kArgOp); in MoveHeadOutsideCompilationToHost()