Searched defs:indice_shape (Results 1 – 6 of 6) sorted by relevance
104 …auto indice_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[0]->GetShape())[kSh… in SparseFillEmptyRowsInferShape() local138 …auto indice_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[0]->GetShape())[kSh… in SparseFillEmptyRowsFrontendInferShape() local
42 …auto indice_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex0]->GetS… in SparseToDenseInferShape() local
85 …std::vector<int64_t> indice_shape = std::vector<int64_t>(inputs[kIndex0]->GetDeviceShapeVector().b… in Resize() local
44 auto indice_shape = common::AnfAlgo::GetPrevNodeOutputInferShape(origin_node, 1); in CheckInputs() local
71 auto indice_shape = inputs[kIndex1]->GetShapeVector(); in Resize() local
249 auto indice_shape = indice->GetTensorShape(); in DoCompute() local