Searched refs:kDeviceArgOp (Results 1 – 7 of 7) sorted by relevance
25 static const char* const kDeviceArgOp = FunctionLibraryDefinition::kDeviceArgOp; variable
88 REGISTER_SYSTEM_KERNEL_BUILDER(Name(kDeviceArgOp).Device(DEVICE_CPU), ArgOp);109 Name(kDeviceArgOp).Device(DEVICE_GPU).TypeConstraint<int32>("T"), ArgOp);
38 n->type_string() == FunctionLibraryDefinition::kDeviceArgOp) { in FunctionBody()
55 static constexpr const char* const kDeviceArgOp = variable56 FunctionLibraryDefinition::kDeviceArgOp;
58 static constexpr const char* const kDeviceArgOp = variable59 FunctionLibraryDefinition::kDeviceArgOp;
47 FunctionLibraryDefinition::kDeviceArgOp; member in tensorflow::FunctionLibraryDefinition194 gnode->set_op(FunctionLibraryDefinition::kDeviceArgOp); in BuildInputArgIndex()615 n->op() == FunctionLibraryDefinition::kDeviceArgOp) { in Print()
371 static constexpr const char* const kDeviceArgOp = "_DeviceArg"; variable