/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()
|