Searched refs:kDeviceArgOp (Results 1 – 5 of 5) sorted by relevance
25 static const char* const kDeviceArgOp = FunctionLibraryDefinition::kDeviceArgOp; variable
72 REGISTER_SYSTEM_KERNEL_BUILDER(Name(kDeviceArgOp).Device(DEVICE_CPU), ArgOp);110 Name(kDeviceArgOp).Device(DEVICE_GPU).TypeConstraint<int32>("T"), ArgOp);
400 static constexpr const char* const kDeviceArgOp = "_DeviceArg"; variable
176 gnode->set_op(FunctionLibraryDefinition::kDeviceArgOp); in BuildInputArgIndex()588 n->op() == FunctionLibraryDefinition::kDeviceArgOp) { in Print()
51 static constexpr const char* const kDeviceArgOp = variable52 FunctionLibraryDefinition::kDeviceArgOp;1932 } else if (n->type_string() == kArgOp || n->type_string() == kDeviceArgOp) { in FunctionBody()