Searched refs:createContiguous (Results 1 – 5 of 5) sorted by relevance
| /external/pytorch/torch/csrc/jit/codegen/fuser/ |
| D | compiler.cpp | 264 auto type = TensorType::createContiguous(*scalar_type, device, sizes); in compileKernel()
|
| /external/pytorch/aten/src/ATen/core/ |
| D | jit_type.h | 608 static TensorTypePtr createContiguous(
1736 return TensorType::createContiguous(at::kLong, at::kCPU, {}); in fromNumberType()
1738 return TensorType::createContiguous(at::kDouble, at::kCPU, {}); in fromNumberType()
1740 return TensorType::createContiguous(at::kBool, at::kCPU, {}); in fromNumberType()
1747 return TensorType::createContiguous(at::kBool, at::kCPU, {}); in fromBoolType()
|
| D | tensor_type.cpp | 448 TensorTypePtr TensorType::createContiguous( in createContiguous() function in c10::TensorType
|
| /external/pytorch/torch/csrc/jit/passes/ |
| D | shape_analysis.cpp | 1966 node->output()->setType(TensorType::createContiguous( in PropagateCompleteShapeOnNode()
2135 TensorType::createContiguous(at::kLong, at::kCPU, dims)); in PropagateCompleteShapeOnNode()
|
| /external/pytorch/torch/csrc/jit/passes/onnx/ |
| D | peephole.cpp | 387 TensorTypePtr newType = TensorType::createContiguous( in pushPackingPastRnn()
|