Searched defs:output_kernel_tensor (Results 1 – 7 of 7) sorted by relevance
41 …DynamicSequenceType(const AnfNodePtr &output_node, const kernel::KernelTensorPtr &output_kernel_te… in UpdateDynamicSequenceType()286 const auto &output_kernel_tensor = AnfAlgo::GetOutputKernelTensor(output_node, output_index); in CreateOutputTensor() local
164 const auto &output_kernel_tensor = output_device_tensor_[0]->kernel_tensor(); in FetchDeviceTensor() local
124 for (auto &output_kernel_tensor : output_kernel_tensors_) { in Init() local1213 for (const auto &output_kernel_tensor : output_kernel_tensors_) { in ProcessMultiStreamAfterKernelLaunch() local
384 const auto &output_kernel_tensor = output_device_address->kernel_tensor(); in CheckOverflow() local
431 const auto &output_kernel_tensor = AnfAlgo::GetOutputKernelTensor(input_node, 0); in UpdateDynamicShapeAndSize() local
334 const auto &output_kernel_tensor = output_addr->kernel_tensor(); in AddArgSpecForInput() local
1298 const auto &output_kernel_tensor = device_tensor->kernel_tensor(); in ConstructOutputByTupleTensor() local