Searched defs:operands_shapes (Results 1 – 1 of 1) sorted by relevance
44 auto operands_shapes = builder->GetOperandShapes(operands).ValueOrDie(); in GetOperandTypes() local114 auto operands_shapes = builder->GetOperandShapes(operands).ValueOrDie(); in AggregateToTopKBuilder() local169 auto operands_shapes = builder->GetOperandShapes(operands).ValueOrDie(); in ApproxTopK() local259 auto operands_shapes = builder->GetOperandShapes(operands).ValueOrDie(); in ApproxTopKFallback() local