Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dfunction_ops.cc71 REGISTER_SYSTEM_KERNEL_BUILDER(Name(kArgOp).Device(DEVICE_CPU), ArgOp);
79 Name(kArgOp).Device(DEVICE_SYCL).TypeConstraint<type>("T"), ArgOp);
81 TF_CALL_bool(REGISTER) REGISTER_KERNEL_BUILDER(Name(kArgOp)
101 Name(kArgOp).Device(DEVICE_GPU).TypeConstraint<type>("T"), ArgOp);
104 TF_CALL_bool(REGISTER) REGISTER_KERNEL_BUILDER(Name(kArgOp)
113 REGISTER_KERNEL_BUILDER(Name(kArgOp)
119 REGISTER_KERNEL_BUILDER(Name(kArgOp)
126 Name(kArgOp).Device(DEVICE_GPU).TypeConstraint<Variant>("T"), ArgOp);
Dfunction_ops.h24 static const char* const kArgOp = FunctionLibraryDefinition::kArgOp; variable
/external/tensorflow/tensorflow/compiler/tf2xla/
Dtf2xla.cc51 const char* const kArgOp = "_Arg"; variable
93 NodeBuilder(absl::StrCat("_arg_", arg_index), kArgOp) in AddArgNodes()
180 if (n->type_string() == kArgOp) { in RewriteAndPruneGraph()
184 return errors::Aborted(kArgOp, in RewriteAndPruneGraph()
211 if (n->type_string() == kArgOp) { in CollectArgNodes()
218 "Multiple ", kArgOp, " nodes with index ", index, ", ", in CollectArgNodes()
226 return errors::InvalidArgument("Expected ", kArgOp, " node with index ", in CollectArgNodes()
Dfunctionalize_while.cc135 const char* const kArgOp = "_Arg"; in BuildArgNode() local
137 NodeDefBuilder builder(absl::StrCat(kArgOp, index), kArgOp); in BuildArgNode()
Dxla_compiler.cc914 NodeBuilder(arg_name, FunctionLibraryDefinition::kArgOp) in CompileSingleOp()
Dfunctionalize_cond.cc431 FunctionLibraryDefinition::kArgOp) in BuildArgumentNodes()
/external/tensorflow/tensorflow/compiler/jit/
Dxla_device_ops.h211 Name(kArgOp).Device(DEVICE).TypeConstraint("T", TYPES), ArgOp); \
212 REGISTER_KERNEL_BUILDER(Name(kArgOp) \
218 Name(kArgOp).Device(DEVICE).TypeConstraint<Variant>("T"), ArgOp); \
273 REGISTER_KERNEL_BUILDER(Name(FunctionLibraryDefinition::kArgOp) \
Dencapsulate_subgraphs_pass.cc138 static const char* const kArgOp = "_Arg"; variable
783 absl::StrCat(src_node->name(), "_", src_slot, "_arg"), kArgOp, in RecordArg()
2498 if (n->type_string() == kArgOp) { in GetArgTypes()
2515 if (n->type_string() == kArgOp) { in RenumberArguments()
Dextract_outside_compilation_pass.cc857 NodeDefBuilder arg_builder("key_arg", FunctionLibraryDefinition::kArgOp); in ReplaceKeyPlaceholderWithArgNode()
/external/tensorflow/tensorflow/core/framework/
Dgraph_to_functiondef.cc35 const char* const kArgOp = "_Arg"; variable
125 if (node->type_string() == kArgOp) { in GraphToFunctionDef()
Dfunction.h399 static constexpr const char* const kArgOp = "_Arg"; variable
Dfunction.cc178 gnode->set_op(FunctionLibraryDefinition::kArgOp); in BuildInputArgIndex()
587 if (n->op() == FunctionLibraryDefinition::kArgOp || in Print()
/external/tensorflow/tensorflow/core/common_runtime/
Dshape_refiner.cc60 constexpr char kArgOp[] = "_Arg"; variable
71 if (StringPiece(node->type_string()) == kArgOp) { in InferShapesForFunctionSubNode()
Dfunction.cc50 static constexpr const char* const kArgOp = FunctionLibraryDefinition::kArgOp; variable
127 ndef.set_op(kArgOp); in AddArg()
846 (n->op_def().is_stateful() && n->type_string() != kArgOp) || in PruneFunctionBody()
1932 } else if (n->type_string() == kArgOp || n->type_string() == kDeviceArgOp) { in FunctionBody()
/external/tensorflow/tensorflow/core/kernels/data/
Ddataset_utils.cc59 if (ret_input_node->def().op() == FunctionLibraryDefinition::kArgOp) { in ComputeShortCircuitIndices()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_graph.cc653 FunctionLibraryDefinition::kArgOp); in RegisterSegmentFunctionToFunctionLibrary()