Searched refs:kDeviceArgOp (Results 1 – 5 of 5) sorted by relevance
25 static const char* const kDeviceArgOp = FunctionLibraryDefinition::kDeviceArgOp; variable
73 REGISTER_SYSTEM_KERNEL_BUILDER(Name(kDeviceArgOp).Device(DEVICE_CPU), ArgOp);111 Name(kDeviceArgOp).Device(DEVICE_GPU).TypeConstraint<int32>("T"), ArgOp);
328 static constexpr const char* const kDeviceArgOp = "_DeviceArg"; variable
47 FunctionLibraryDefinition::kDeviceArgOp; member in tensorflow::FunctionLibraryDefinition194 gnode->set_op(FunctionLibraryDefinition::kDeviceArgOp); in BuildInputArgIndex()615 n->op() == FunctionLibraryDefinition::kDeviceArgOp) { in Print()
55 static constexpr const char* const kDeviceArgOp = variable56 FunctionLibraryDefinition::kDeviceArgOp;2357 } else if (n->type_string() == kArgOp || n->type_string() == kDeviceArgOp) { in FunctionBody()