Searched refs:set_host_shape (Results 1 – 2 of 2) sorted by relevance
81 void set_host_shape(const ShapeVector &shape) { host_shape_ = shape; } in set_host_shape() function
236 device_address->set_host_shape(trans::GetRuntimePaddingShape(node, i)); in RunOpMallocPre()474 device_address->set_host_shape(trans::GetRuntimePaddingShape(kernel, i)); in RunOpAssignOutputMemory()928 device_address->set_host_shape(trans::GetRuntimePaddingShape(node, i)); in AssignNodeOutputMem()