Searched refs:SetOperandBytesAccessed (Results 1 – 2 of 2) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | hlo_cost_analysis.cc | 62 SetOperandBytesAccessed(i, GetShapeSize(operand->shape())); in Preprocess() 235 SetOperandBytesAccessed(0, 0); in HandleGetTupleElement() 255 SetOperandBytesAccessed(0, GetShapeSize(slice->shape())); in HandleSlice() 265 SetOperandBytesAccessed(0, GetShapeSize(dynamic_slice->shape())); in HandleDynamicSlice() 266 SetOperandBytesAccessed(1, GetShapeSize(dynamic_slice->operand(1)->shape())); in HandleDynamicSlice() 278 SetOperandBytesAccessed(0, 0); in HandleDynamicUpdateSlice() 279 SetOperandBytesAccessed( in HandleDynamicUpdateSlice() 281 SetOperandBytesAccessed( in HandleDynamicUpdateSlice() 294 SetOperandBytesAccessed(i, 0); in HandleTuple() 317 SetOperandBytesAccessed(i, 0); in HandleDomain() [all …]
|
D | hlo_cost_analysis.h | 230 void SetOperandBytesAccessed(int64 operand_num, float value); 231 void SetOperandBytesAccessed(int64 operand_num, ShapeIndex index,
|