Searched refs:dim_callback_ (Results 1 – 1 of 1) sorted by relevance
502 dim_callback_(dim_callback), in CAPICustomDeviceTensorHandle()514 *dim = dim_callback_(dim_index, &s); in Dim()521 DimCallback dim_callback_; member in tensorflow::__anon6f1bef310211::CAPICustomDeviceTensorHandle