Searched defs:tensorinfo_in (Results 1 – 3 of 3) sorted by relevance
336 TensorInfoBasePtr tensorinfo_in; in GetTensorInLayoutForNewShape() local374 auto tensorinfo_in = in GetTensorInLayout() local381 auto tensorinfo_in = distribute_operator->outputs_tensor_info()[0]; in GetTensorInLayout() local1621 … TensorInfo tensorinfo_in = distribute_operator->inputs_tensor_info()[LongToSize(res.second - 1)]; in SetParallelShape() local1624 TensorInfoBasePtr tensorinfo_in; in SetParallelShape() local
613 TensorInfo tensorinfo_in = distribute_operator->inputs_tensor_info()[LongToSize(res.second - 1)]; in SetParallelShape() local
2823 TensorInfo tensorinfo_in = distribute_operator->inputs_tensor_info()[LongToSize(index - 1)]; in GetInputLayoutFromCNode() local2830 auto tensorinfo_in = distribute_operator->inputs_tensor_info_new()[LongToSize(index - 1)]; in GetInputLayoutFromCNode() local