| /external/tensorflow/tensorflow/core/kernels/ |
| D | function_ops.cc | 72 REGISTER_SYSTEM_KERNEL_BUILDER(Name(kArgOp).Device(DEVICE_CPU), ArgOp);
80 Name(kArgOp).Device(DEVICE_SYCL).TypeConstraint<type>("T"), ArgOp);
82 TF_CALL_bool(REGISTER) REGISTER_KERNEL_BUILDER(Name(kArgOp)
102 Name(kArgOp).Device(DEVICE_GPU).TypeConstraint<type>("T"), ArgOp);
105 TF_CALL_bool(REGISTER) REGISTER_KERNEL_BUILDER(Name(kArgOp)
114 REGISTER_KERNEL_BUILDER(Name(kArgOp)
120 REGISTER_KERNEL_BUILDER(Name(kArgOp)
127 Name(kArgOp).Device(DEVICE_GPU).TypeConstraint<Variant>("T"), ArgOp);
|
| D | function_ops.h | 24 static const char* const kArgOp = FunctionLibraryDefinition::kArgOp; variable
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | graph_compiler_util.cc | 82 FunctionLibraryDefinition::FunctionLibraryDefinition::kArgOp) in AddArgNodes()
170 if (n->type_string() == FunctionLibraryDefinition::kArgOp) { in RewriteAndPruneGraph()
174 return errors::Aborted(FunctionLibraryDefinition::kArgOp, in RewriteAndPruneGraph()
201 if (n->type_string() == FunctionLibraryDefinition::kArgOp) { in CollectArgNodes()
208 "Multiple ", FunctionLibraryDefinition::kArgOp, in CollectArgNodes()
218 "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 | 439 FunctionLibraryDefinition::kArgOp) in BuildArgumentNodes()
|
| D | xla_compiler.cc | 1172 NodeBuilder(arg_name, FunctionLibraryDefinition::kArgOp) in CompileSingleOp()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_device_ops.h | 144 Name(kArgOp).Device(DEVICE).TypeConstraint("T", TYPES), ArgOp); \
145 REGISTER_KERNEL_BUILDER(Name(kArgOp) \
151 Name(kArgOp).Device(DEVICE).TypeConstraint<Variant>("T"), ArgOp); \
212 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()
1135 if (n->type_string() == kArgOp) { in GetArgTypes()
1152 if (n->type_string() == kArgOp) { in RenumberArguments()
|
| D | extract_outside_compilation_pass.cc | 1338 NodeDefBuilder arg_builder("key_arg", FunctionLibraryDefinition::kArgOp); in ReplaceKeyPlaceholderWithArgNode()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | shape_refiner.cc | 60 constexpr char kArgOp[] = "_Arg"; variable
71 if (StringPiece(node->type_string()) == kArgOp) { in InferShapesForFunctionSubNode()
|
| D | function.cc | 54 static constexpr const char* const kArgOp = FunctionLibraryDefinition::kArgOp; variable
127 ndef.set_op(kArgOp); in AddArg()
914 (n->op_def().is_stateful() && n->type_string() != kArgOp) || in PruneFunctionBody()
2357 } else if (n->type_string() == kArgOp || n->type_string() == kDeviceArgOp) { in FunctionBody()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | function.h | 327 static constexpr const char* const kArgOp = "_Arg";
|
| 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()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | export_graphdef.cc | 251 node_def->set_op(FunctionLibraryDefinition::kArgOp); in GetArgumentNode()
|
| D | import_model.cc | 663 op_name != FunctionLibraryDefinition::kArgOp) { in AddNodesToShapeRefiner()
749 if (StringPiece(node->type_string()) == FunctionLibraryDefinition::kArgOp) { in AddNodesToShapeRefiner()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | captured_function.cc | 178 if (ret_input_node->def().op() == FunctionLibraryDefinition::kArgOp) { in CreateShortCircuitInfo()
|