Searched defs:tensor_desc (Results 1 – 16 of 16) sorted by relevance
30 TensorDescriptor tensor_desc) { in CreatePReLU()
139 const TensorDescriptor& tensor_desc, in CheckExternalTensorDescription()207 TensorDescriptor tensor_desc; in ReserveGraphTensors() local729 const TensorDescriptor& tensor_desc) in ResolvePolymorphicArgs()750 const auto& tensor_desc = gpu_model->tensors[node.inputs[i]]; in ResolvePolymorphicArgs() local757 const auto& tensor_desc = gpu_model->tensors[node.outputs[i]]; in ResolvePolymorphicArgs() local
119 const TensorDescriptor &tensor_desc = tensor.desc(); in create_tensor() local
167 const TensorDescriptor &tensor_desc = tensor.desc(); in create_tensor() local
1571 TensorDescriptor tensor_desc = in CreateBhwcTensorDescriptor() local1580 TensorDescriptor tensor_desc = in CreateHwcTensorDescriptor() local1606 TensorDescriptor tensor_desc = in CreateConstantLinearTensorDescriptor() local1624 TensorDescriptor tensor_desc = in CreateConstantHWVec4TensorDescriptor() local
553 const auto* tensor_desc = in Encode() local580 const auto* tensor_desc = in Encode() local
344 TensorDescriptor* tensor_desc; in ResolveSecondElementwiseInput() local
541 const auto* tensor_desc = dynamic_cast<const TensorDescriptor*>(desc_ptr); in ResolveSelector() local
565 const auto& tensor_desc = gpu_model.tensors.at(tensor_id); in AllocateBufferBasedTensors() local617 TensorDescriptor tensor_desc; in AllocateStrongShapesTensors() member647 const auto& tensor_desc = gpu_model.tensors.at(tensor_id); in AllocateStrongShapesTensors() local
333 const auto* tensor_desc = dynamic_cast<const TensorDescriptor*>(obj_ptr); in GetGPUResources() local
113 const auto* tensor_desc = dynamic_cast<const TensorDescriptor*>(desc); in CreateCLObject() local
215 const auto* tensor_desc = dynamic_cast<const TensorDescriptor*>(obj_ptr); in GetGPUResources() local
74 const auto* tensor_desc = dynamic_cast<const TensorDescriptor*>(desc); in CreateMetalObject() local
608 TensorDescriptor tensor_desc; in AllocateMemoryForStrongShapes() member
1660 TensorDescriptor tensor_desc = CreateTensorDescriptor(); in Create() local1679 TensorDescriptor tensor_desc = CreateTensorDescriptor(); in Create() local
384 uint64_t GetHashValue(miopenTensorDescriptor_t tensor_desc) { in GetHashValue()