Searched defs:t_dtype (Results 1 – 4 of 4) sorted by relevance
15 vkapi::ScalarType t_dtype, in TensorRef()
38 inline aotriton::DType cast_dtype(caffe2::TypeMeta t_dtype) in cast_dtype()
235 DataType t_dtype = DataTypeToEnum<T>::v(); in RunFusedBatchNormEx() local361 DataType t_dtype = DataTypeToEnum<T>::v(); in VerifyTensorsNear() local
1663 DataType t_dtype = GetDataTypeFromAttr(*fused_batch_norm_node_def, "T"); in FindFusedBatchNormEx() local1804 DataType t_dtype = GetDataTypeFromAttr(*node_def, "T"); in FindFusedBatchNormGradEx() local