Searched refs:expanded_tensor (Results 1 – 2 of 2) sorted by relevance
56 Tensor expanded_tensor = self.expand(padded_size);63 auto inputDataType = getMPSDataType(expanded_tensor);69 …or* inputTensor = mpsGraphRankedPlaceHolder(mpsGraph, inputDataType, getMPSShape(expanded_tensor));77 …cachedGraph->inputTensor_, expanded_tensor, /*mpsShape=*/nil, /*gatherTensorData=*/true, inputData…
3708 ITensorProxyPtr expanded_tensor = nullptr; in ConvertPack() local3717 /*output=*/&expanded_tensor, in ConvertPack()3726 /*tensor=*/&expanded_tensor, in ConvertPack()3731 expanded_tensors.push_back(expanded_tensor); in ConvertPack()