Searched refs:SetOutputBytesAccessed (Results 1 – 2 of 2) sorted by relevance
58 SetOutputBytesAccessed(GetShapeSize(hlo->shape())); in Preprocess()211 SetOutputBytesAccessed(0); in HandleParameter()219 SetOutputBytesAccessed(0); in HandleConstant()234 SetOutputBytesAccessed(0); in HandleGetTupleElement()254 SetOutputBytesAccessed(GetShapeSize(slice->shape())); in HandleSlice()264 SetOutputBytesAccessed(GetShapeSize(dynamic_slice->shape())); in HandleDynamicSlice()276 SetOutputBytesAccessed( in HandleDynamicUpdateSlice()292 SetOutputBytesAccessed(GetShapeSize(tuple->shape())); in HandleTuple()315 SetOutputBytesAccessed(0); in HandleDomain()344 SetOutputBytesAccessed(indexed_shape.index, in HandleInfeed()[all …]
235 void SetOutputBytesAccessed(float value);236 void SetOutputBytesAccessed(ShapeIndex index, float value);