Home
last modified time | relevance | path

Searched refs:SetOperandBytesAccessed (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_cost_analysis.cc62 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 …]
Dhlo_cost_analysis.h230 void SetOperandBytesAccessed(int64 operand_num, float value);
231 void SetOperandBytesAccessed(int64 operand_num, ShapeIndex index,