Searched defs:device_shape (Results 1 – 6 of 6) sorted by relevance
252 std::vector<size_t> device_shape; in NhwcDeviceShape() local264 std::vector<int64_t> device_shape; in NhwcDeviceDynamicShape() local276 std::vector<size_t> device_shape; in HwchDeviceShape() local288 std::vector<int64_t> device_shape; in HwchDeviceDynamicShape() local300 std::vector<size_t> device_shape; in FracZDeviceShape() local315 std::vector<int64_t> device_shape; in FracZDeviceDynamicShape() local337 std::vector<size_t> device_shape; in Nc1hwc0DeviceShape() local352 std::vector<int64_t> device_shape; in Nc1hwc0DeviceDynamicShape() local369 std::vector<size_t> device_shape; in Ndc1hwc0DeviceShape() local387 std::vector<int64_t> device_shape; in Ndc1hwc0DeviceDynamicShape() local[all …]
67 std::vector<size_t> device_shape; member
51 …static ShapeVector GetFakeAbstractShape(const ShapeVector &device_shape, const std::string &format… in GetFakeAbstractShape()69 static ShapeVector NchwAbstractShape(const ShapeVector &device_shape) { return device_shape; } in NchwAbstractShape()70 static ShapeVector NhwcAbstractShape(const ShapeVector &device_shape) { in NhwcAbstractShape()76 static ShapeVector FractalNzAbstractShape(const ShapeVector &device_shape) { in FractalNzAbstractShape()332 ShapeVector GetFakeAbstractShape(const ShapeVector &device_shape, const std::string &format) { in GetFakeAbstractShape()
243 std::vector<size_t> device_shape; in GetDeviceShape() local318 std::vector<size_t> device_shape = GetDeviceShape(&host_shape); in SyncDeviceToHostAndConvertFormat() local441 std::vector<size_t> device_shape; in ConvertFormatAndSyncHostToDevice() local
170 auto device_shape = GetDeviceShape(scalar_tensor_node); in ReplaceTensorWithScalar() local640 auto device_shape = AnfAlgo::GetOutputDeviceShape(node, 0); in GetDeviceShape() local
2129 auto device_shape = GetInputDeviceShape(anf_node, index); in GetInputRealDeviceShapeIfExist() local2141 auto device_shape = GetOutputDeviceShape(anf_node, index); in GetOutputRealDeviceShapeIfExist() local