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,
|