| /external/tensorflow/tensorflow/core/kernels/ |
| D | function_ops.cc | 71 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);
|
| D | function_ops.h | 24 static const char* const kArgOp = FunctionLibraryDefinition::kArgOp; variable
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | tf2xla.cc | 51 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()
|
| D | functionalize_while.cc | 135 const char* const kArgOp = "_Arg"; in BuildArgNode() local
137 NodeDefBuilder builder(absl::StrCat(kArgOp, index), kArgOp); in BuildArgNode()
|
| D | xla_compiler.cc | 914 NodeBuilder(arg_name, FunctionLibraryDefinition::kArgOp) in CompileSingleOp()
|
| D | functionalize_cond.cc | 431 FunctionLibraryDefinition::kArgOp) in BuildArgumentNodes()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_device_ops.h | 211 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) \
|
| D | encapsulate_subgraphs_pass.cc | 138 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()
|
| D | extract_outside_compilation_pass.cc | 857 NodeDefBuilder arg_builder("key_arg", FunctionLibraryDefinition::kArgOp); in ReplaceKeyPlaceholderWithArgNode()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | graph_to_functiondef.cc | 35 const char* const kArgOp = "_Arg"; variable
125 if (node->type_string() == kArgOp) { in GraphToFunctionDef()
|
| D | function.h | 399 static constexpr const char* const kArgOp = "_Arg"; variable
|
| D | function.cc | 178 gnode->set_op(FunctionLibraryDefinition::kArgOp); in BuildInputArgIndex()
587 if (n->op() == FunctionLibraryDefinition::kArgOp || in Print()
|
| /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 | 50 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/ |
| D | dataset_utils.cc | 59 if (ret_input_node->def().op() == FunctionLibraryDefinition::kArgOp) { in ComputeShortCircuitIndices()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_graph.cc | 653 FunctionLibraryDefinition::kArgOp); in RegisterSegmentFunctionToFunctionLibrary()
|