Searched refs:output_tensor_size (Results 1 – 3 of 3) sorted by relevance
52 std::vector<size_t> output_tensor_size) in OpRuntimeInfo() argument55 output_tensor_size_(std::move(output_tensor_size)) {} in OpRuntimeInfo()63 size_t output_tensor_size(size_t index) const;
2497 size_t OpRuntimeInfo::output_tensor_size(size_t index) const { in output_tensor_size() function in mindspore::session::OpRuntimeInfo
188 …CreateDeviceAddress(nullptr, runtime_info->output_tensor_size(i), output_format, output_type, {nod… in RunOpGetCommunicationOutputInfo()232 auto tensor_size = runtime_info->output_tensor_size(i); in RunOpMallocPre()264 auto output_tensor_size = op_runtime_info->output_tensor_size(index); in RunOpMallocPre() local267 … CreateDeviceAddress(nullptr, output_tensor_size, output_format, output_type_id, {item, index}); in RunOpMallocPre()