Searched defs:output_device_shape (Results 1 – 2 of 2) sorted by relevance
63 std::vector<ShapeVector> output_device_shape; in CacheForExecutionOrder() local108 std::vector<ShapeVector> output_device_shape; in CacheForGraphInputs() local159 const ShapeVector &OpRuntimeInfo::output_device_shape(size_t index) const { in output_device_shape() function in mindspore::runtime::OpRuntimeInfo
89 std::vector<ShapeVector> output_device_shape, device::KernelInfo *kernel_info, in OpRuntimeInfo()