| /external/tensorflow/tensorflow/core/kernels/ |
| function_ops.h | 24 static const char* const kArgOp = FunctionLibraryDefinition::kArgOp;
|
| 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);
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| tf2xla.cc | 51 const char* const kArgOp = "_Arg";
93 NodeBuilder(absl::StrCat("_arg_", arg_index), kArgOp)
180 if (n->type_string() == kArgOp) {
184 return errors::Aborted(kArgOp,
211 if (n->type_string() == kArgOp) {
218 "Multiple ", kArgOp, " nodes with index ", index, ", ",
226 return errors::InvalidArgument("Expected ", kArgOp, " node with index ",
|
| functionalize_while.cc | 135 const char* const kArgOp = "_Arg";
137 NodeDefBuilder builder(absl::StrCat(kArgOp, index), kArgOp);
|
| functionalize_cond.cc | 431 FunctionLibraryDefinition::kArgOp)
[all...] |
| xla_compiler.cc | [all...] |
| /external/tensorflow/tensorflow/compiler/jit/ |
| 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) \
|
| encapsulate_subgraphs_pass.cc | 138 static const char* const kArgOp = "_Arg";
[all...] |
| extract_outside_compilation_pass.cc | 857 NodeDefBuilder arg_builder("key_arg", FunctionLibraryDefinition::kArgOp)
[all...] |
| /external/tensorflow/tensorflow/core/framework/ |
| graph_to_functiondef.cc | 35 const char* const kArgOp = "_Arg";
125 if (node->type_string() == kArgOp) {
|
| function.h | 397 // Ops created for function arguments bear the name given by `kArgOp`; those
399 static constexpr const char* const kArgOp = "_Arg";
[all...] |
| function.cc | 178 gnode->set_op(FunctionLibraryDefinition::kArgOp);
587 if (n->op() == FunctionLibraryDefinition::kArgOp ||
[all...] |
| /external/tensorflow/tensorflow/core/kernels/data/ |
| dataset_utils.cc | 59 if (ret_input_node->def().op() == FunctionLibraryDefinition::kArgOp) {
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| shape_refiner.cc | 60 constexpr char kArgOp[] = "_Arg";
71 if (StringPiece(node->type_string()) == kArgOp) {
|
| function.cc | 50 static constexpr const char* const kArgOp = FunctionLibraryDefinition::kArgOp;
127 ndef.set_op(kArgOp);
846 (n->op_def().is_stateful() && n->type_string() != kArgOp) ||
[all...] |
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| convert_graph.cc | 653 FunctionLibraryDefinition::kArgOp);
[all...] |
