Searched refs:kIntsOnDeviceAttr (Results 1 – 5 of 5) sorted by relevance
436 if (fdef.attr().count(FunctionLibraryDefinition::kIntsOnDeviceAttr) != 0 && in ValidateMultiDeviceOptions()437 fdef.attr().at(FunctionLibraryDefinition::kIntsOnDeviceAttr).b()) { in ValidateMultiDeviceOptions()440 FunctionLibraryDefinition::kIntsOnDeviceAttr, in ValidateMultiDeviceOptions()
403 static constexpr const char* const kIntsOnDeviceAttr = variable
691 fdef.attr().count(FunctionLibraryDefinition::kIntsOnDeviceAttr) != 0 && in InstantiateFunction()692 fdef.attr().at(FunctionLibraryDefinition::kIntsOnDeviceAttr).b(); in InstantiateFunction()
550 (*fdef.mutable_attr())[FunctionLibraryDefinition::kIntsOnDeviceAttr].set_b( in TEST()
719 ->mutable_attr())[FunctionLibraryDefinition::kIntsOnDeviceAttr] in RegisterSegmentFunctionToFunctionLibrary()