Searched refs:kIntsOnDeviceAttr (Results 1 – 6 of 6) sorted by relevance
294 fdef->attr().count(FunctionLibraryDefinition::kIntsOnDeviceAttr) != 0 && in ConstructFunctionHandle()295 fdef->attr().at(FunctionLibraryDefinition::kIntsOnDeviceAttr).b(); in ConstructFunctionHandle()
513 auto attr = fdef->attr().find(FunctionLibraryDefinition::kIntsOnDeviceAttr); in Create()516 << FunctionLibraryDefinition::kIntsOnDeviceAttr << " attribute."; in Create()
604 if (fdef.attr().count(FunctionLibraryDefinition::kIntsOnDeviceAttr) != 0 && in ValidateMultiDeviceOptions()605 fdef.attr().at(FunctionLibraryDefinition::kIntsOnDeviceAttr).b()) { in ValidateMultiDeviceOptions()608 FunctionLibraryDefinition::kIntsOnDeviceAttr, in ValidateMultiDeviceOptions()
52 FunctionLibraryDefinition::kIntsOnDeviceAttr; member in tensorflow::FunctionLibraryDefinition727 fdef.attr().count(FunctionLibraryDefinition::kIntsOnDeviceAttr) != 0 && in InstantiateFunction()728 fdef.attr().at(FunctionLibraryDefinition::kIntsOnDeviceAttr).b(); in InstantiateFunction()
374 static constexpr const char* const kIntsOnDeviceAttr = variable
553 (*fdef.mutable_attr())[FunctionLibraryDefinition::kIntsOnDeviceAttr].set_b( in TEST()