Searched refs:outputIndicesShape (Results 1 – 1 of 1) sorted by relevance
79 Shape outputIndicesShape; in prepare() local80 outputIndicesShape.type = OperandType::TENSOR_INT32; in prepare()81 outputIndicesShape.dimensions = inputShape.dimensions; in prepare()82 outputIndicesShape.dimensions.back() = k; in prepare()84 context->setOutputShape(kOutputIndicesTensor, outputIndicesShape); in prepare()