Searched refs:GetTypeByte (Results 1 – 13 of 13) sorted by relevance
110 size_t type_size = GetTypeByte(TypeIdToType(output_type_id)); in AssignValueNodeAddress()143 size_t type_size = GetTypeByte(TypeIdToType(output_type_id)); in AssignInputNodeAddress()209 size_t type_size = GetTypeByte(TypeIdToType(device_type_id)); in CreatTensorForOutput()227 size_t type_size = GetTypeByte(TypeIdToType(device_type_id)); in CreatTensorForOutput()332 …if (GetTypeByte(TypeIdToType(tensor->data_type())) == GetTypeByte(TypeIdToType(address->type_id_))… in BindInputTensorAddressPtr()337 … GetTypeByte(TypeIdToType(address->type_id_)), std::multiplies<size_t>()); in BindInputTensorAddressPtr()
32 type_size_ = GetTypeByte(TypeIdToType(x_data_type_)); in InitKernel()
35 input_x_dtype_size_ = GetTypeByte(TypeIdToType(input_x_dtype_)); in InitKernel()
31 type_size_ = GetTypeByte(TypeIdToType(input_x_dtype_)); in InitKernel()
33 size_t type_size = GetTypeByte(TypeIdToType(type_id)); in InitInputOutputSize()43 size_t type_size = GetTypeByte(TypeIdToType(type_id)); in InitInputOutputSize()
651 size_t mem_size = GetTypeByte(tensor_type_ptr) * IntToSize(new_tensor_ptr->ElementsNum());691 auto tensor_type_byte = GetTypeByte(tensor_type_ptr);722 size_t mem_size = GetTypeByte(tensor_type_ptr) * IntToSize(new_tensor_ptr->ElementsNum());774 size_t mem_size = GetTypeByte(tensor_type_ptr) * IntToSize(new_tensor_ptr->ElementsNum());907 … size_t mem_size = GetTypeByte(new_tensor_ptr->Dtype()) * IntToSize(new_tensor_ptr->ElementsNum());
48 size_t GetTypeByte(const TypePtr &type_ptr);
168 size_t GetTypeByte(const TypePtr &type_ptr) { in GetTypeByte() function
360 size_t type_size = GetTypeByte(TypeIdToType(output_type)); in CreateFusedCommunicationOp()
72 size_t type_size = GetTypeByte(TypeIdToType(ou_type)); in CalculOriInput()
400 size_t mem_size = GetTypeByte(tensor_type_ptr) * IntToSize(new_tensor_ptr->ElementsNum()); in operator()
912 size_t type_byte = GetTypeByte(type_ptr); in GetShapeSize()
646 size_t type_size = GetTypeByte(TypeIdToType(output_type_id)); in GetOutputTensorMemSize()